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

..

__init__.py

Improved docs for pytorch_lightning.core (continued) (#1483)

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

decorators.py

add rank warning (#1428)

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

grads.py

Type Hints for Lightning Core (#946)

2020-03-12 12:47:23 -04:00

hooks.py

support for native amp (#1561)

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

lightning.py

support for native amp (#1561)

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

memory.py

Improved docs for pytorch_lightning.core (continued) (#1483)

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

model_saving.py

add rank warning (#1428)

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

root_module.py

add rank warning (#1428)

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

saving.py

Improved docs for pytorch_lightning.core (continued) (#1483)

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