pretrain 1
Browse files
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 |
+
```
|