pretrain core 4
Browse files
scripts/pretrain_core_model_4.yaml
CHANGED
|
@@ -47,12 +47,12 @@ data:
|
|
| 47 |
|
| 48 |
init_args:
|
| 49 |
data_path: "../core-data-4-8193-16385-16385-1000/"
|
| 50 |
-
num_workers:
|
| 51 |
|
| 52 |
# Training-related arguments. See ``litgpt.args.TrainArgs`` for details
|
| 53 |
train:
|
| 54 |
# Number of optimizer steps between saving checkpoints (type: Optional[int], default: 1000)
|
| 55 |
-
save_interval:
|
| 56 |
|
| 57 |
# Number of iterations between logging calls (type: int, default: 1)
|
| 58 |
log_interval: 1
|
|
|
|
| 47 |
|
| 48 |
init_args:
|
| 49 |
data_path: "../core-data-4-8193-16385-16385-1000/"
|
| 50 |
+
num_workers: 1
|
| 51 |
|
| 52 |
# Training-related arguments. See ``litgpt.args.TrainArgs`` for details
|
| 53 |
train:
|
| 54 |
# Number of optimizer steps between saving checkpoints (type: Optional[int], default: 1000)
|
| 55 |
+
save_interval: 10
|
| 56 |
|
| 57 |
# Number of iterations between logging calls (type: int, default: 1)
|
| 58 |
log_interval: 1
|