Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

scaled batch size

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/models/trainer.py
View File

@ -136,7 +136,7 @@ class Trainer(TrainerIO):
if self.use_amp: if self.use_amp:
print('using 16bit precision') print('using 16bit precision')
if self.use_amp and not APEX_AVAILABLE: if use_amp and not APEX_AVAILABLE:
msg = ''' msg = '''
You set use_amp=True but do not have apex installed. You set use_amp=True but do not have apex installed.
Install apex first using this guide and rerun with use_amp=True: Install apex first using this guide and rerun with use_amp=True: