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
|