protomotions.agents.callbacks.slurm_autoresume_srun module¶

protomotions.agents.callbacks.slurm_autoresume_srun.wandb_run_exists()[source]¶

class protomotions.agents.callbacks.slurm_autoresume_srun.AutoResumeCallbackSrun(autoresume_after=12600)[source]¶

Bases: <Mock object at 0x7ff4232fcd10>[]

__init__(autoresume_after=12600)[source]¶

before_play_steps(agent)[source]¶

on_fit_start(agent)[source]¶

on_fit_end(agent)[source]¶