change save limit to 1
Browse files- scripts/train.py +1 -1
scripts/train.py
CHANGED
|
@@ -79,7 +79,7 @@ def main():
|
|
| 79 |
logging_dir="./logs",
|
| 80 |
logging_steps=10,
|
| 81 |
save_steps=20,
|
| 82 |
-
save_total_limit=
|
| 83 |
bf16=True,
|
| 84 |
gradient_checkpointing=False, # Disabled for now
|
| 85 |
report_to="tensorboard",
|
|
|
|
| 79 |
logging_dir="./logs",
|
| 80 |
logging_steps=10,
|
| 81 |
save_steps=20,
|
| 82 |
+
save_total_limit=1,
|
| 83 |
bf16=True,
|
| 84 |
gradient_checkpointing=False, # Disabled for now
|
| 85 |
report_to="tensorboard",
|