mtasic85 commited on
Commit
ef18151
·
1 Parent(s): 7688300

pretrain 1

Browse files
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -214,3 +214,7 @@ CUDA_VISIBLE_DEVICES=0 CUDA_LAUNCH_BLOCKING=0 PYTORCH_CUDA_ALLOC_CONF=expandable
214
  ```bash
215
  litgpt convert_pretrained_checkpoint ../out/pretrain-core-0/final ../out/pretrain-core-0/checkpoint
216
  ```
 
 
 
 
 
214
  ```bash
215
  litgpt convert_pretrained_checkpoint ../out/pretrain-core-0/final ../out/pretrain-core-0/checkpoint
216
  ```
217
+
218
+ ```bash
219
+ CUDA_VISIBLE_DEVICES=0 CUDA_LAUNCH_BLOCKING=0 PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True litgpt pretrain --config pretrain_core_model_1.yaml
220
+ ```