Skip to content

Commit

Permalink
cleaned max_model_t
Browse files Browse the repository at this point in the history
  • Loading branch information
Michael Janner authored Aug 6, 2019
1 parent 1c2c84a commit 22cab51
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion mbpo/algorithms/mbpo.py
Original file line number Diff line number Diff line change
Expand Up @@ -221,7 +221,6 @@ def _train(self):
self._epoch, self._model_train_freq, self._timestep, self._total_timestep, self._train_steps_this_epoch, self._num_train_steps)
)

# max_t = self._max_model_t if self._epoch > 100 else 1e10
model_train_metrics = self._train_model(batch_size=256, max_epochs=None, holdout_ratio=0.2, max_t=self._max_model_t)
model_metrics.update(model_train_metrics)
gt.stamp('epoch_train_model')
Expand Down

0 comments on commit 22cab51

Please sign in to comment.