four4fish
|
15cd6ad45b
|
Call TrainingTypePlugin collective functions directly instead of going through the Accelerator (#9677)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: ananthsub <ananth.subramaniam@gmail.com>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
|
2021-09-27 14:52:57 +02:00 |
|
Jirka Borovec
|
6e124e7207
|
CI: precommit - docformatter (#8584)
* CI: precommit - docformatter
* fix deprecated
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2021-09-06 12:49:09 +00:00 |
|
Carlos Mocholí
|
b1a859f312
|
Remove deprecated `on_{save,load}_checkpoint` signature (#8697)
Co-authored-by: Yifu Wang <yifuwang2012@gmail.com>
|
2021-08-21 22:48:28 -07:00 |
|
Carlos Mocholí
|
a64cc37394
|
Replace `yapf` with `black` (#7783)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
|
2021-07-26 13:37:35 +02:00 |
|
Adrian Wälchli
|
7e6010fc93
|
fix info message when max training time reached (#7780)
* call time_elapsed
* elapsed formatting
* format
* update test
* changelog
|
2021-05-31 14:50:16 +02:00 |
|
Adrian Wälchli
|
67d21609c9
|
Add Trainer max_time argument + Callback (#6823)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
|
2021-04-16 13:38:57 +02:00 |