refactored model tests

tests/debug.py

@@ -111,7 +111,7 @@ def main():
         max_nb_epochs=1,
         gpus=[0, 1],
         distributed_backend='dp',
-        use_amp=True
+        use_amp=False
     )
 
     result = trainer.fit(model)