3 people 29ebe92208 support for native amp (#1561) ...

* adding native amp suppport

* adding native amp suppport

* adding native amp suppport

* adding native amp suppport

* autocast

* autocast

* autocast

* autocast

* autocast

* autocast

* removed comments

* removed comments

* added state saving

* added state saving

* try install amp again

* added state saving

* drop Apex reinstall

Co-authored-by: J. Borovec <jirka.borovec@seznam.cz>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

2020-04-23 14:47:08 -04:00

..

callbacks

Default value for ModelCheckpoint filepath (#1548)

2020-04-23 11:50:58 -04:00

core

support for native amp (#1561)

2020-04-23 14:47:08 -04:00

loggers

Improved docs for Loggers (#1484)

2020-04-16 12:04:12 -04:00

logging

add rank warning (#1428)

2020-04-09 14:05:46 -04:00

overrides

Add ddp_cpu backend for testing ddp without GPUs (#1158)

2020-04-15 23:17:31 -04:00

profiler

Profiler summary (#1259)

2020-03-31 08:57:48 -04:00

pt_overrides

add rank warning (#1428)

2020-04-09 14:05:46 -04:00

root_module

add rank warning (#1428)

2020-04-09 14:05:46 -04:00

trainer

support for native amp (#1561)

2020-04-23 14:47:08 -04:00

utilities

Revert namespace package search to normal package search (#1545)

2020-04-21 08:26:47 -04:00

__init__.py

Update __init__.py

2020-04-23 11:04:57 -04:00