Skip to content
Snippets Groups Projects
Commit 36f58292 authored by Risto Luukkonen's avatar Risto Luukkonen
Browse files

fix typo

parent e5acde6a
No related branches found
No related tags found
No related merge requests found
...@@ -96,7 +96,7 @@ if __name__ == '__main__': ...@@ -96,7 +96,7 @@ if __name__ == '__main__':
MODEL_CHECKPOINT = cmd_args.from_checkpoint MODEL_CHECKPOINT = cmd_args.from_checkpoint
EPOCHS = cmd_args.epochs EPOCHS = cmd_args.epochs
BATCH_SIZE = cmd_args.batch_size BATCH_SIZE = cmd_args.batch_size
GRD_ACC_STEPS = cmd_args.grad_acc GRAD_ACC_STEPS = cmd_args.grad_acc
tokenizer_path = '/projappl/project_2004600/risto/tokenizer/' tokenizer_path = '/projappl/project_2004600/risto/tokenizer/'
MODEL = 'gpt2-xl' MODEL = 'gpt2-xl'
DS_CONF = cmd_args.deepspeed_config DS_CONF = cmd_args.deepspeed_config
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment