William Falcon
|
bcd13f70b8
|
ref: run_pretrain_routine -> setup_training (#3294)
* ref: .tune()
* ref: run_pretrain_routine -> setup_training
|
2020-08-31 18:06:11 -04:00 |
|
William Falcon
|
6bae404bed
|
ref: ddp backend refactor (3) (#3208)
* ddp backend refactor
* ddp backend refactor
|
2020-08-26 20:03:09 -04:00 |
|
William Falcon
|
82d1128966
|
eval step scaling factor (#3136)
|
2020-08-24 20:26:39 -04:00 |
|
William Falcon
|
6c3cec3a3c
|
training amp scaling refactor (#3135)
|
2020-08-24 19:59:46 -04:00 |
|
William Falcon
|
0b3cb3c955
|
ref: moved ___step_end hooks (#3130)
* moved eval hooks
* moved eval hooks
* moved eval hooks
* moved eval hooks
* moved eval hooks
* moved eval hooks
* moved eval hooks
|
2020-08-24 17:50:47 -04:00 |
|
William Falcon
|
6068b29d29
|
ref: remove obscure forward call in eval + CPU backend ___step (#3123)
* remove obscure forward call in eval
* remove obscure forward call in eval
* remove obscure forward call in eval
* remove obscure forward call in eval
* remove obscure forward call in eval
* remove obscure forward call in eval
|
2020-08-24 12:31:40 -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 |
|
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
|
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 |