Rares Dolga

Results 1 issues of Rares Dolga

Hi, I am trying to save checkpoints using the following code: ``` options = ocp.CheckpointManagerOptions( max_to_keep=self.max_checkpoints, create=True, best_fn=best_loss, best_mode="min", ) self.checkpoint_manager = ocp.CheckpointManager( os.path.join(self._out_dir, "checkpoints"), options=options, item_names=("state", "metadata"), item_handlers={ "state":...