o vžÄiZã@sHddlmZddlZddlZddlmZddlmZGdd„deƒZdS)é)ÚOptionalN)ÚAbstractCheckpointer)ÚImaginaireModelc @s€eZdZdZdedejjdejjj dej j de ddf d d „Z   d dedeejjdeejjj deej j de f d d „ZdS)Ú CheckpointerzŠ A dummy checkpointer that does not save or load anything. This is useful for debugging jobs or share workload with collobrators. ÚmodelÚ optimizerÚ schedulerÚ grad_scalerÚ iterationÚreturnNcCsdS)N©)Úselfrrrr r r r úV/data/cameron/vidgen/cosmos-policy/cosmos_policy/_src/imaginaire/checkpointer/dummy.pyÚsaveszCheckpointer.savecCsdS)Nrr )r rrrr r r rÚload(szCheckpointer.load)NNN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚtorchÚoptimÚ OptimizerÚ lr_schedulerÚ LRSchedulerÚampÚ GradScalerÚintrrrr r r rrs:þýüûú ù ûþ ý ü ûúr) ÚtypingrrÚtorch.distributedÚ/cosmos_policy._src.imaginaire.checkpointer.baserÚ#cosmos_policy._src.imaginaire.modelrrr r r rÚs