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

scaled batch size

Browse Source
This commit is contained in:
William Falcon
2019-07-08 20:02:06 -04:00
parent 49ad7d6c28
commit 51c55c938a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+1 -1
View File
@@ -136,7 +136,7 @@ class Trainer(TrainerIO):
if self.use_amp:
print('using 16bit precision')
if self.use_amp and not APEX_AVAILABLE:
if use_amp and not APEX_AVAILABLE:
msg = '''
You set use_amp=True but do not have apex installed.
Install apex first using this guide and rerun with use_amp=True: