src/transformers/trainer.py · d5bc32ce92ace9aaec7752e0b89d51ba18903a1b · chenpangpang / transformers

Trainer + wandb quality of life logging tweaks (#6241) · bd0eab35

Teven authored Aug 05, 2020



* added `name` argument for wandb logging, also logging model config with trainer arguments

* Update src/transformers/training_args.py
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* added tf, post-review changes
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

bd0eab35

trainer.py 44.9 KB

Replace trainer.py