Skip to content

Commit 576fe73

Browse files
author
James Danielson
committed
More readme clarification
1 parent 9730cff commit 576fe73

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

README.md

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,14 @@ You can use it like so: `--model gpt2-xl`
2424
### Train - Run, run, run...
2525

2626
Train for X epochs using input.model and save to output.model Then train again for more epochs until coherent.
27-
`out/input.model` and `out/output.model` can be the same model to resume and continue training.
27+
--model `out/output.model` and `out/output.model` should be the same model to resume and continue training.
28+
29+
If you wish to save a newly trained model to a new `out/output2.model/` path you should copy the tokenized output from the first step into the new output directory
2830

2931
Note: change `--batch-size` for smaller/larger GPUs, default is 4.
3032

3133
```bash
32-
python train.py --model [out/input.model] --epochs X [out/output.model]
34+
python train.py --model [out/output.model] --epochs X [out/output.model]
3335
```
3436

3537
# Generate text with the model

0 commit comments

Comments
 (0)