SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add tests to Trainer (#6605)

Browse Source 
* Add tests to Trainer

* Test if removing long breaks everything

* Remove ugly hack

* Fix distributed test

* Use float for number of epochs
This commit is contained in:
Sylvain Gugger
2020-08-20 11:13:50 -04:00
committed by GitHub
parent 039d8d65fc
commit 573bdb0a5d
6 changed files with 313 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/training_args.py
View File

@@ -69,7 +69,8 @@ class TrainingArguments:
max_grad_norm (:obj:`float`, `optional`, defaults to 1.0):
Maximum gradient norm (for gradient clipping).
num_train_epochs(:obj:`float`, `optional`, defaults to 3.0):
Total number of training epochs to perform.
Total number of training epochs to perform (if not an integer, will perform the decimal part percents of
the last epoch before stopping training).
max_steps (:obj:`int`, `optional`, defaults to -1):
If set to a positive number, the total number of training steps to perform. Overrides
:obj:`num_train_epochs`.