This commit is contained in:
Suraj Patil
2021-07-16 16:44:59 +05:30
committed by GitHub
parent c07334c12e
commit 8ef3f36561
3 changed files with 1 additions and 1 deletions

View File

@@ -52,7 +52,7 @@ model.save_pretrained("gpt-neo-1.3B")
### Train Model ### Train Model
```bash ```bash
python run_pjit_clm.py \ python run_clm_mp.py \
--model_name_or_path gpt-neo-1.3B \ --model_name_or_path gpt-neo-1.3B \
--tokenizer_name gpt2 \ --tokenizer_name gpt2 \
--dataset_name wikitext --dataset_config_name wikitext-2-raw-v1 \ --dataset_name wikitext --dataset_config_name wikitext-2-raw-v1 \