mtasic85 commited on
Commit
46f1dc6
1 Parent(s): 1d1adae
Files changed (1) hide show
  1. scripts/model.yaml +1 -1
scripts/model.yaml CHANGED
@@ -32,7 +32,7 @@ precision: bf16-true
32
 
33
  # Optional path to a checkpoint directory to initialize the model from.
34
  # Useful for continued pretraining. Mutually exclusive with ``resume``. (type: Optional[Path], default: null)
35
- initial_checkpoint_dir: "../out/pretrain/initial/"
36
 
37
  # Path to a checkpoint directory to resume from in case training was interrupted, or ``True`` to resume
38
  # from the latest checkpoint in ``out_dir``. An error will be raised if no checkpoint is found. Passing
 
32
 
33
  # Optional path to a checkpoint directory to initialize the model from.
34
  # Useful for continued pretraining. Mutually exclusive with ``resume``. (type: Optional[Path], default: null)
35
+ initial_checkpoint_dir: "../out/pretrain/pretrained_checkpoint_converted"
36
 
37
  # Path to a checkpoint directory to resume from in case training was interrupted, or ``True`` to resume
38
  # from the latest checkpoint in ``out_dir``. An error will be raised if no checkpoint is found. Passing