Update oumi/oumi_train.yaml
Browse files- oumi/oumi_train.yaml +1 -1
oumi/oumi_train.yaml
CHANGED
|
@@ -59,7 +59,7 @@ training:
|
|
| 59 |
log_model_summary: False
|
| 60 |
empty_device_cache_steps: 1
|
| 61 |
include_performance_metrics: True
|
| 62 |
-
output_dir: "output/
|
| 63 |
enable_wandb: True
|
| 64 |
|
| 65 |
fsdp:
|
|
|
|
| 59 |
log_model_summary: False
|
| 60 |
empty_device_cache_steps: 1
|
| 61 |
include_performance_metrics: True
|
| 62 |
+
output_dir: "output/llama8b.sft"
|
| 63 |
enable_wandb: True
|
| 64 |
|
| 65 |
fsdp:
|