William Falcon
|
527b9dca36
|
refactored ddp backend forward (#3119)
|
2020-08-24 07:33:14 -04:00 |
|
Jirka Borovec
|
4354690e55
|
add apex test (#2921)
* add apex test
* rename
* level
* events
* wrap
* evt
* miss
* apex
* apex
* apex
* apex
* apex
* apex
* Update tests/models/test_amp.py
Co-authored-by: William Falcon <waf2107@columbia.edu>
* notes
* notes
Co-authored-by: William Falcon <waf2107@columbia.edu>
|
2020-08-13 10:03:13 -04:00 |
|
Phil
|
e3528afae3
|
Move optimizer creation after device placement for ddp backends. (#2904)
|
2020-08-12 06:34:59 -04:00 |
|
Jirka Borovec
|
a6e7aa7796
|
allow using apex with any PT version (#2865)
* wip
* setup
* type
* name
* wip
* docs
* imports
* fix if
* fix if
* use_amp
* Apply suggestions from code review
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
* Apply suggestions from code review
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
* fix tests
* Apply suggestions from code review
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
* fix tests
* todos
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
|
2020-08-08 11:07:32 +02:00 |
|
Jirka Borovec
|
b7d72706c3
|
clean imports (#2867)
* clean imports
* miss
|
2020-08-08 00:33:51 +02:00 |
|
Jirka Borovec
|
f8c058215f
|
simplify tests & cleaning (#2588)
* simplify
* tmpdir
* revert
* clean
* accel
* types
* test
* edit test acc
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
* Update test acc
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
|
2020-08-07 23:22:05 +02:00 |