From 88fbf6cc4b1e8d6b6fb27478ec22ffc2b65fa13c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 13 May 2019 20:44:25 -0400 Subject: [PATCH] added option to change default tensor --- pytorch_lightning/models/trainer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 9cafea88..aa8074c5 100644 --- 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: