Update changelog

araffin · Oct 27, 2018 · dde1afa · dde1afa
1 parent 44c14f9
commit dde1afa
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/docs/changelog.rst b/docs/changelog.rst
@@ -5,6 +5,12 @@ Changelog
 
 For download links, please look at `Github release page <https://github.com/araffin/robotics-rl-srl/releases>`_.
 
+Pre-Release 1.0.1 (WIP)
+-----------------------
+
+- fixed a bug in the dataset generator where the GUI was instantiated two times
+- updated stable-baselines version + srl-zoo submodule
+
 Release 1.0 (2018-10-09)
 -------------------------
 **Stable Baselines Version**

diff --git a/environments/dataset_generator.py b/environments/dataset_generator.py
@@ -62,11 +62,13 @@ def env_thread(args, thread_num, partition=True, use_ppo2=False):
 
     env_class = registered_env[args.env][0]
     env = env_class(**env_kwargs)
+
+    # Additional env when using a trained ppo agent to generate data
+    # instead of a random agent
     train_env = env_class(**{**env_kwargs, "record_data": False, "renders": False})
     train_env = DummyVecEnv([lambda: train_env])
     train_env = VecNormalize(train_env, norm_obs=True, norm_reward=False)
 
-
     model = None
     if use_ppo2:
         model = PPO2(CnnPolicy, train_env).learn(args.ppo2_timesteps)