train model
Browse files- scripts/train_model.py +1 -0
scripts/train_model.py
CHANGED
|
@@ -233,6 +233,7 @@ training_args = TrainingArguments(
|
|
| 233 |
logging_steps=10,
|
| 234 |
fp16=False,
|
| 235 |
bf16=True,
|
|
|
|
| 236 |
)
|
| 237 |
print(training_args)
|
| 238 |
|
|
|
|
| 233 |
logging_steps=10,
|
| 234 |
fp16=False,
|
| 235 |
bf16=True,
|
| 236 |
+
torch_compile=True,
|
| 237 |
)
|
| 238 |
print(training_args)
|
| 239 |
|