transformers
22fc93c4 - Changes in create_optimizer to support tensor parallelism with SMP (#16880)

Commit
3 years ago
Changes in create_optimizer to support tensor parallelism with SMP (#16880) * changes in create optimizer to support tensor parallelism with SMP * Update src/transformers/trainer.py Convert if check to one line. Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Co-authored-by: Cavdar <dcavdar@a07817b12d7e.ant.amazon.com> Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Author
Parents
Loading