diff --git a/safe_opax/rl/trainer.py b/safe_opax/rl/trainer.py index 5f1d2bc2..ea4c202c 100644 --- a/safe_opax/rl/trainer.py +++ b/safe_opax/rl/trainer.py @@ -185,7 +185,6 @@ def _run_training_epoch( def from_pickle(cls, config: DictConfig, state_path: str) -> "Trainer": with open(state_path, "rb") as f: make_env, seeds, agent, epoch, step = cloudpickle.load(f).values() - assert agent.config == config, "Loaded different hyperparameters." _LOG.info(f"Resuming from step {step}") return cls( config=agent.config,