From a8f3b1b21fca5b2cd62bbe10e7c432fc857efd40 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Mon, 8 Jul 2019 20:03:08 -0400
Subject: [PATCH] scaled batch size

---
 pytorch_lightning/models/trainer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 69d4fcd4..a85fbafe 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -13,6 +13,7 @@ import os
 import subprocess
 from time import sleep
 from torch.utils.data.distributed import DistributedSampler
+import warnings
 
 
 try:
@@ -142,8 +143,7 @@ class Trainer(TrainerIO):
             Install apex first using this guide and rerun with use_amp=True: 
             https://github.com/NVIDIA/apex#linux
             '''
-            raise Warning(msg)
-
+            warnings.warn(msg)
             print('not using 16 bit')
 
     def __determine_data_use_amount(self, train_percent_check, val_percent_check, test_percent_check, overfit_pct):