added option to change default tensor

2019-05-13 20:44:25 -04:00 · 2019-05-13 20:44:25 -04:00 · 88fbf6cc4b
parent 7002de1d4e
commit 88fbf6cc4b
1 changed files with 2 additions and 0 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -5,8 +5,10 @@ from pytorch_lightning.root_module.memory import get_gpu_memory_map
 import traceback
 from pytorch_lightning.root_module.model_saving import TrainerIO
 from torch.optim.lr_scheduler import MultiStepLR
+import pdb

 try:
+    pdb.set_trace()
    from apex import amp
    APEX_AVAILABLE = True
 except ModuleNotFoundError: