mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
60d4d5e1aa
* Remove all __future__ imports from RLlib. * Remove (object) again from tf_run_builder.py::TFRunBuilder. * Fix 2xLINT warnings. * Fix broken appo_policy import (must be appo_tf_policy) * Remove future imports from all other ray files (not just RLlib). * Remove future imports from all other ray files (not just RLlib). * Remove future import blocks that contain `unicode_literals` as well. Revert appo_tf_policy.py to appo_policy.py (belongs to another PR). * Add two empty lines before Schedule class. * Put back __future__ imports into determine_tests_to_run.py. Fails otherwise on a py2/print related error.
32 lines
965 B
Python
32 lines
965 B
Python
import gym
|
|
import time
|
|
import unittest
|
|
|
|
import ray
|
|
from ray.rllib.evaluation.rollout_worker import RolloutWorker
|
|
from ray.rllib.tests.test_rollout_worker import MockPolicy
|
|
|
|
|
|
class TestPerf(unittest.TestCase):
|
|
# Tested on Intel(R) Core(TM) i7-4600U CPU @ 2.10GHz
|
|
# 11/23/18: Samples per second 8501.125113727468
|
|
# 03/01/19: Samples per second 8610.164353268685
|
|
def testBaselinePerformance(self):
|
|
for _ in range(20):
|
|
ev = RolloutWorker(
|
|
env_creator=lambda _: gym.make("CartPole-v0"),
|
|
policy=MockPolicy,
|
|
batch_steps=100)
|
|
start = time.time()
|
|
count = 0
|
|
while time.time() - start < 1:
|
|
count += ev.sample().count
|
|
print()
|
|
print("Samples per second {}".format(
|
|
count / (time.time() - start)))
|
|
print()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
ray.init(num_cpus=5)
|
|
unittest.main(verbosity=2)
|