four4fish
|
6e14209185
|
Rewrite accelerator_connector (#11448)
|
2022-02-17 23:38:39 +00:00 |
|
Rohit Gupta
|
9ed44dee0d
|
Fix to avoid moving batch to device for DataParallel (#11780)
Co-authored-by: ananthsub <ananth.subramaniam@gmail.com>
|
2022-02-07 14:26:18 +00:00 |
|
Rohit Gupta
|
7ec1e66e17
|
reduce only loss with dp (#11594)
Co-authored-by: Aki Nitta <nitta@akihironitta.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
|
2022-02-07 17:00:29 +05:30 |
|
ananthsub
|
f41d1e5e5e
|
Remove `Strategy.on_gpu` (#11537)
|
2022-01-19 21:27:12 +00:00 |
|
Kaushik B
|
576a5d62a0
|
Introduce strategies directory for Training Strategies (#11226)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
|
2021-12-22 20:23:30 +00:00 |