890588a9ee
ref: precision plugins 1/n ( #3504 )
...
* ref: precision plugins 1/n
* ref: precision plugins 1/n
2020-09-15 09:56:12 -04:00
6bcfa8b068
ref: merge backends x/n ( #3482 )
2020-09-12 16:28:29 -04:00
00d155ae01
ref: merge backends x/n ( #3477 )
2020-09-12 12:36:55 -04:00
dd324e4086
ref: accelerator connector methods x/n ( #3470 )
2020-09-11 22:25:48 -04:00
ef20310873
ref: move specific accelerator code x/n ( #3457 )
...
* ref: organize args x/n
* ref: move specific accelerator code x/n
* ref: move specific accelerator code x/n
* ref: move specific accelerator code x/n
2020-09-11 10:56:21 -04:00
70af47db84
ref: organize args 4/n ( #3456 )
2020-09-10 21:58:47 -04:00
8f6b115511
ref: added model connector ( #3407 )
...
* ref: added model connector
* ref: added model connector
* ref: added model connector
2020-09-09 00:24:20 -04:00
b0298cead8
ref: move train outside of setup training ( #3297 )
...
* ref: move train outside of setup training
* ref: move train outside of setup training
* ref: move train outside of setup training
* ref: move train outside of setup training
2020-08-31 20:36:52 -04:00
bcd13f70b8
ref: run_pretrain_routine -> setup_training ( #3294 )
...
* ref: .tune()
* ref: run_pretrain_routine -> setup_training
2020-08-31 18:06:11 -04:00
3a26b4ff5c
ddp backend refactor ( #3209 )
2020-08-26 20:31:09 -04:00
6bae404bed
ref: ddp backend refactor (3) ( #3208 )
...
* ddp backend refactor
* ddp backend refactor
2020-08-26 20:03:09 -04:00
a8daf914f8
ddp backend refactor ( #3207 )
2020-08-26 19:10:24 -04:00
ff3c2f4cff
ddp backend refactor ( #3204 )
2020-08-26 18:43:28 -04:00
6c3cec3a3c
training amp scaling refactor ( #3135 )
2020-08-24 19:59:46 -04:00
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
527b9dca36
refactored ddp backend forward ( #3119 )
2020-08-24 07:33:14 -04:00
188e06c261
ddp fix for trainer.test() + add basic ddp tests ( #2997 )
...
* add ddp script variations
* add ddp test
* rename
* shell
* test
* test
* try call
* try without subprocess
* test
* display the error
* list all variations
* try string
* try copy env
* debug
* pythonpath
* path
* update test
* change
* simple ddp test
* replace
* remove random port
* random port
* str
* clean up
* check run spawn
* clean up
* docs
* docs
* update test
* docs
* changelog
* changelog
2020-08-16 11:19:57 -04:00
e7794eb79a
Fixes #2407 ( #2981 )
...
* fix gpus index error
2020-08-14 16:22:48 -04:00
5bce06c050
nb. devices ( #2973 )
2020-08-14 11:37:21 +02:00
0c264689cb
Fixes #2942 ( #2969 )
...
* Fixes #2942
* doc fix
2020-08-13 21:54:57 -04:00
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
e3528afae3
Move optimizer creation after device placement for ddp backends. ( #2904 )
2020-08-12 06:34:59 -04:00
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
b7d72706c3
clean imports ( #2867 )
...
* clean imports
* miss
2020-08-08 00:33:51 +02:00
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
William Falcon