wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Set precision=16 when use_amp is passed as True (#1145)

Browse Source 
* Set precision=16 when use_amp is passed as True

* Update CHANGELOG.md

* add use_amp to deprecated API

* Update trainer.py

* Update trainer.py

* move the use_amp attribute to deprecated API

* move use_amp deprecation back to Trainer's __init__

* drop unsed

* drop deprecated

* reorder imports

* typing

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: William Falcon <waf2107@columbia.edu>
Co-authored-by: J. Borovec <jirka.borovec@seznam.cz>
This commit is contained in:
Roshan Rao
2020-04-06 08:13:24 -04:00
committed by GitHub
co-authored by Jirka Borovec William Falcon J. Borovec
parent 26cb5f6817
commit 4ed3027309
8 changed files with 34 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pl_examples/basic_examples/gpu_template.py
+1 -1
View File
@@ -32,7 +32,7 @@ def main(hparams):
max_epochs=hparams.epochs,
gpus=hparams.gpus,
distributed_backend=hparams.distributed_backend,
use_amp=hparams.use_16bit
precision=16 if hparams.use_16bit else 32,
)
# ------------------------