# Save only best model in Trainer

**URL:** </static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fsave-only-best-model-in-trainer%2F8442%3E%5C
**Category:** 🤗Transformers\
**Created:** [July 17, 2021, 6:25am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fsave-only-best-model-in-trainer%2F8442 "2021-07-17T06:25:08Z")\
**Posts on this page:** 1\
**Showing post:** 4

<div class="post-metadata">

**Author:** ![prachi12](https://avatars.discourse-cdn.com/v4/letter/p/b2d939/32.png) [@prachi12](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fprachi12)%5C
**Post date:** [July 17, 2021, 4:22pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fsave-only-best-model-in-trainer%2F8442%2F4 "2021-07-17T16:22:43Z")

</div>

I am using the below -

```auto
  args = TrainingArguments(
      output_dir=f"./out_fold{i}",
      overwrite_output_dir = 'True',
      evaluation_strategy="steps",
      eval_steps=40,
      logging_steps = 40,
      learning_rate = 5e-5,
      per_device_train_batch_size=8,
      per_device_eval_batch_size=8,
      num_train_epochs=10,
      seed=0,
      save_total_limit = 1,
      # report_to = "none",
  # logging_steps = 'epoch',
      load_best_model_at_end=True,
      save_strategy = "no"
  )
  trainer = Trainer(
      model=model,
      args=args,
      train_dataset=train_dataset,
      eval_dataset=val_dataset,
      # compute_metrics=compute_metrics,
      
      # callbacks=[EarlyStoppingCallback(early_stopping_pa)],
  )
  trainer.train()
  trainer.save_model(f'out_fold{i}')

```

Here, thought save\_strategy = “no” , the checkpoints are saved at start in disk (as below) due to which disk goes full. Can you suggest what’s going wrong?

\*\*\*\*\* Running Evaluation \*\*\*\*\*  
Num examples = 567  
Batch size = 8  
Saving model checkpoint to ./out\_fold0/checkpoint-40  
Configuration saved in ./out\_fold0/checkpoint-40/config.json  
Model weights saved in ./out\_fold0/checkpoint-40/pytorch\_model.bin  
Deleting older checkpoint [out\_fold0/checkpoint-760] due to args.save\_total\_limit  
\*\*\*\*\* Running Evaluation \*\*\*\*\*

---

_[View the full topic](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fsave-only-best-model-in-trainer%2F8442)._