diff --git a/docker/examples/Dockerfile b/docker/examples/Dockerfile index 5f17d31c5..c9a964d8a 100644 --- a/docker/examples/Dockerfile +++ b/docker/examples/Dockerfile @@ -4,4 +4,4 @@ FROM ray-project/deploy RUN conda install -y -c conda-forge tensorflow RUN apt-get install -y zlib1g-dev RUN pip install gym[atari] opencv-python==3.2.0.8 smart_open -RUN conda install -y -q pytorch torchvision -c soumith +# RUN conda install -y -q pytorch torchvision -c soumith diff --git a/python/ray/rllib/utils/timer.py b/python/ray/rllib/utils/timer.py index 3b692f3ab..98770b671 100644 --- a/python/ray/rllib/utils/timer.py +++ b/python/ray/rllib/utils/timer.py @@ -24,9 +24,9 @@ class TimerStat(RunningStat): def __enter__(self): assert self._start_time is None, "concurrent updates not supported" - self._start_time = time.monotonic() + self._start_time = time.time() def __exit__(self, type, value, tb): assert self._start_time is not None - self.push(time.monotonic() - self._start_time) + self.push(time.time() - self._start_time) self._start_time = None diff --git a/test/jenkins_tests/run_multi_node_tests.sh b/test/jenkins_tests/run_multi_node_tests.sh index 1789dbf5e..ff5269f79 100755 --- a/test/jenkins_tests/run_multi_node_tests.sh +++ b/test/jenkins_tests/run_multi_node_tests.sh @@ -139,12 +139,12 @@ docker run --rm --shm-size=10G --memory=10G $DOCKER_SHA \ --stop '{"training_iteration": 2}' \ --config '{"kl_coeff": 1.0, "num_sgd_iter": 10, "sgd_stepsize": 1e-4, "sgd_batchsize": 64, "timesteps_per_batch": 2000, "num_workers": 1, "model": {"dim": 40, "conv_filters": [[16, [8, 8], 4], [32, [4, 4], 2], [512, [5, 5], 1]]}, "extra_frameskip": 4}' -docker run --rm --shm-size=10G --memory=10G $DOCKER_SHA \ - python /ray/python/ray/rllib/train.py \ - --env PongDeterministic-v4 \ - --run A3C \ - --stop '{"training_iteration": 2}' \ - --config '{"num_workers": 2, "use_lstm": false, "use_pytorch": true, "model": {"grayscale": true, "zero_mean": false, "dim": 80, "channel_major": true}}' +# docker run --rm --shm-size=10G --memory=10G $DOCKER_SHA \ +# python /ray/python/ray/rllib/train.py \ +# --env PongDeterministic-v4 \ +# --run A3C \ +# --stop '{"training_iteration": 2}' \ +# --config '{"num_workers": 2, "use_lstm": false, "use_pytorch": true, "model": {"grayscale": true, "zero_mean": false, "dim": 80, "channel_major": true}}' docker run --rm --shm-size=10G --memory=10G $DOCKER_SHA \ python /ray/python/ray/rllib/test/test_checkpoint_restore.py