mtasic85 commited on
Commit
7dbb613
·
1 Parent(s): 69919ad

pretrain core 4

Browse files
Files changed (1) hide show
  1. scripts/pretrain_core_model_4.yaml +2 -2
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: 32
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: 1
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