src/transformers/integrations.py/NeptuneCallback/on_save
class NeptuneCallback:    def on_save(self, args, state, control, **kwargs):
        if self._should_upload_checkpoint:
            self._log_model_checkpoint(args.output_dir, f"checkpoint-{state.global_step}")