stable-baselines3/stable_baselines3/common
Bernhard Raml 15e94a6d14
Add support to log videos via tensorboard (#196)
* Add support to log videos via tensorboard

The ability to look at renderings of agent's trajectories during
training helps evaluate the performance of that agent. One can see what
the agent actually does at various stages during training. For now only
tensorboard is supported, as it is straightforward to implement.

* Remove moviepy dependency from extra & doc update

* Removed the moviepy dependency from the `extra` dependencies so the
user can decide whether to install it or not

* Update the video logging docu with proper naming, comments

* Added a warning to the video logging docu explaining the moviepy
dependency

* Updated the video test, to check for a warning when moviepy is missing

* Update doc

* Update FormatUnsupportedError message

* Also log the offending value making the error message more expressive

* Fix reporting the correct format and update regression test

* Use string description in FormatUnsupportedError

* Instead of converting the value to string without the user's control
the constructor takes a string representation of the value

* Use string description in FormatUnsupportedError

* Use a shorter string description for the error to reduce verbosity

Co-authored-by: Bernhard Raml <raml.bernhard@gmail.com>
Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2020-10-22 11:33:58 +02:00
..
sb2_compat Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
vec_env Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
__init__.py Update docs (custom policy, type hints) (#167) 2020-09-29 20:41:14 +03:00
atari_wrappers.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
base_class.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
bit_flipping_env.py Cleanup docstring types (#169) 2020-10-02 20:05:55 +03:00
buffers.py Fix off-by-one GAE computation (#185) 2020-10-13 00:10:54 +03:00
callbacks.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
cmd_util.py Rename cmd_util to env_util (#197) 2020-10-22 11:05:52 +02:00
distributions.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
env_checker.py add check to ensure action space is non-dict non-tuple for env_checker nan check (#192) 2020-10-19 00:23:51 +03:00
env_util.py Rename cmd_util to env_util (#197) 2020-10-22 11:05:52 +02:00
evaluation.py Cleanup docstring types (#169) 2020-10-02 20:05:55 +03:00
identity_env.py Cleanup docstring types (#169) 2020-10-02 20:05:55 +03:00
logger.py Add support to log videos via tensorboard (#196) 2020-10-22 11:33:58 +02:00
monitor.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
noise.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
off_policy_algorithm.py Cleanup docstring types (#169) 2020-10-02 20:05:55 +03:00
on_policy_algorithm.py Fix off-by-one GAE computation (#185) 2020-10-13 00:10:54 +03:00
policies.py Fix initializing CUDA even when device="cpu" is used. (#194) 2020-10-18 20:51:56 +02:00
preprocessing.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
results_plotter.py Cleanup docstring types (#169) 2020-10-02 20:05:55 +03:00
running_mean_std.py Cleanup docstring types (#169) 2020-10-02 20:05:55 +03:00
save_util.py Improve typing coverage (#175) 2020-10-07 10:51:49 +02:00
torch_layers.py Add custom arch for off-policy actor/critic networks (#182) 2020-10-13 12:01:33 +02:00
type_aliases.py Update docs (custom policy, type hints) (#167) 2020-09-29 20:41:14 +03:00
utils.py Add custom arch for off-policy actor/critic networks (#182) 2020-10-13 12:01:33 +02:00