Fix logging with multiple seeds (#142)

fix the issue brought up by #106
rail-berkeley · Sep 10, 2021 · e15a945 · e15a945
1 parent 6a13e1b
commit e15a945
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/rlkit/core/batch_rl_algorithm.py b/rlkit/core/batch_rl_algorithm.py
@@ -40,6 +40,7 @@ def __init__(
         self.num_train_loops_per_epoch = num_train_loops_per_epoch
         self.num_expl_steps_per_train_loop = num_expl_steps_per_train_loop
         self.min_num_steps_before_training = min_num_steps_before_training
+        gt.reset_root()
 
     def _train(self):
         if self.min_num_steps_before_training > 0: