[rllib] switch to python logger (#3098)

* logg

* set rllib logger

* comment

* info

* rlib

* comment

* add format

* fix lint

* add file info

* update

* add ts

* lint

* better docs

* fix value error

* soft log level
This commit is contained in:
Eric Liang
2018-10-21 23:43:57 -07:00
committed by GitHub
parent 40c4148d4f
commit 221d1663c1
25 changed files with 160 additions and 320 deletions
@@ -3,11 +3,14 @@ from __future__ import division
from __future__ import print_function
import ray
import logging
from ray.rllib.optimizers.policy_optimizer import PolicyOptimizer
from ray.rllib.evaluation.sample_batch import SampleBatch
from ray.rllib.utils.filter import RunningStat
from ray.rllib.utils.timer import TimerStat
logger = logging.getLogger(__name__)
class SyncSamplesOptimizer(PolicyOptimizer):
"""A simple synchronous RL optimizer.
@@ -52,7 +55,7 @@ class SyncSamplesOptimizer(PolicyOptimizer):
if "stats" in fetches:
self.learner_stats = fetches["stats"]
if self.num_sgd_iter > 1:
print(i, fetches)
logger.debug("{} {}".format(i, fetches))
self.grad_timer.push_units_processed(samples.count)
self.num_steps_sampled += samples.count