Skip to content

Instantly share code, notes, and snippets.

@HeenaR17
Created December 21, 2020 17:15
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save HeenaR17/c692099e4f7e36d0304fadc733ac3e15 to your computer and use it in GitHub Desktop.
Save HeenaR17/c692099e4f7e36d0304fadc733ac3e15 to your computer and use it in GitHub Desktop.
model_args = {
"reprocess_input_data": True,
"overwrite_output_dir": True,
"max_seq_length": 128,
"train_batch_size": 16,
"num_train_epochs": 25,
"num_beams": None,
"do_sample": True,
"max_length": 20,
"top_k": 50,
"top_p": 0.95,
"use_multiprocessing": False,
"save_steps": -1,
"save_eval_checkpoints": True,
"evaluate_during_training": True,
"evaluate_during_training_verbose": True,
"num_return_sequences": 5
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment