wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-04 13:00:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

default to O1 (#334)

Browse Source
This commit is contained in:
William Falcon authored and GitHub committed 2019-10-08 09:09:57 -04:00
1 parent 1cf2e228ba
commit ff2a21a08a
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/trainer.py
+1 -1
View File
@@ -86,7 +86,7 @@ class Trainer(TrainerIO):
print_nan_grads=False,
print_weights_summary=True,
weights_save_path=None,
amp_level='O2',
amp_level='O1',
nb_sanity_val_steps=5):
"""