From 8224a41ba904cd49dfdb262ab40fd68510c396d0 Mon Sep 17 00:00:00 2001 From: Shangtong Zhang Date: Mon, 1 Jan 2018 22:49:19 -0700 Subject: [PATCH] Update README --- README.md | 4 +--- utils/tf_logger.py | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 449288f..f1e9cf3 100644 --- a/README.md +++ b/README.md @@ -94,9 +94,7 @@ Prediction is sampled after 110K iterations and I only implemented one-step trai * [Roboschool](https://github.com/openai/roboschool) (Optional) * PyTorch v0.3.0 * Python 2.7 or Python 3.6 -* Tensorflow (Optional, but tensorboard is awesome) -> If you want to use Roboschool, you have to use Python3. And don't try to use Roboschool with parallelized algorithms, -> there is a known [critical bug](https://github.com/openai/roboschool/issues/86). +* [TensorboardX](https://github.com/lanpa/tensorboard-pytorch) # Usage diff --git a/utils/tf_logger.py b/utils/tf_logger.py index 600dd0a..72e2905 100644 --- a/utils/tf_logger.py +++ b/utils/tf_logger.py @@ -22,4 +22,4 @@ class Logger(object): def histo_summary(self, tag, values, step): if self.skip: return - self.writer.add_histogram(tag, values, step) \ No newline at end of file + self.writer.add_histogram(tag, values, step, bins=1000) \ No newline at end of file