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 |
+
```
|