scaled batch size

2019-07-08 20:02:06 -04:00 · 2019-07-08 20:02:06 -04:00 · 51c55c938a
parent 49ad7d6c28
commit 51c55c938a
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -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: