No hparams check

yardenas · Sep 1, 2024 · 2454f59 · 2454f59
1 parent 52da358
commit 2454f59
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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,