From 8a6846671074106bffe17eb53e4fbed3ca42f46b Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 13 May 2019 21:27:01 -0400 Subject: [PATCH] added option to change default tensor --- pytorch_lightning/models/trainer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 7f7324b917..d15b86f02e 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -225,7 +225,6 @@ class Trainer(TrainerIO): self.model, self.optimizers[0], opt_level="O2", keep_batchnorm_fp32=True, loss_scale="dynamic" ) - pdb.set_trace() self.optimizers[0] = optimizer model.trainer = self @@ -369,7 +368,6 @@ 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: