mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
dd70720578
* bulk rename * deprecation warn * update doc * update fig * line length * rename * make pytest comptaible * fix test * fi sys * rename * wip * fix more * lint * update svg * comments * lint * fix use of batch steps
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
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):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
ray.init(num_cpus=5)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
ray.shutdown()
|
|
|
|
# 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 test_baseline_performance(self):
|
|
for _ in range(20):
|
|
ev = RolloutWorker(
|
|
env_creator=lambda _: gym.make("CartPole-v0"),
|
|
policy=MockPolicy,
|
|
rollout_fragment_length=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__":
|
|
import pytest
|
|
import sys
|
|
sys.exit(pytest.main(["-v", __file__]))
|