Disable checkpointing in Trainer
Hi folks,When I am running a lot of quick and dirty experiments, I do not want / need to save any models as I’m usually relying on the metrics from theTrainerto guide my next decision.One thing that slows down my iteration speed is the fact that the Trainer will save a checkpoint after some number of steps, defined by thesave_stepsparameter inTrainingArguments.To disable checkpointing, what I currently do is setsave_stepsto some large number, but my question is whether there is a more elegant way to do this? For example, is there a Trainer argument I can set that will disable checkpointing altogether?Thanks!
2020-11-30T14:04:38Z
There is none for now. We could definitely add asave_strategylike  theevaluation_strategythat could take the values no/steps/epoch.If you want to tackle this in a PR that would be a welcome contribution!
2020-11-30T14:18:10Z
Thanks for the info! Sure I am happy to tackle this in a PR - will report back when it’s ready
2022-01-02T09:54:15Z
Is this solved now? Is there any way to disable checkpointing or do we have to set thesave_stepsto a large number to disable checkpointing even now?
2022-01-10T15:45:37Z
You can usesave_strategy="no".