From 5f0a71c41402ba2e8e8bee905ed919b3f7d10bfe Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 13 May 2019 21:18:17 -0400 Subject: [PATCH] added option to change default tensor --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index aa8074c533..079df2d02f 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -8,7 +8,6 @@ from torch.optim.lr_scheduler import MultiStepLR import pdb try: - pdb.set_trace() from apex import amp APEX_AVAILABLE = True except ModuleNotFoundError: @@ -369,6 +368,7 @@ class Trainer(TrainerIO): self.__add_tqdm_metrics(model_specific_tqdm_metrics_dic) # backward pass + pdb.set_trace() if self.use_amp: for optimizer in self.optimizers: with amp.scale_loss(loss, optimizer) as scaled_loss: