From 4dbf38093ae319fada3bcac1fd20ec41d94c060a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 13 May 2019 21:22:50 -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 546b2324..7f7324b9 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -219,13 +219,13 @@ class Trainer(TrainerIO): # filter out the weights that were done on gpu so we can load on good old cpus self.optimizers = model.configure_optimizers() - pdb.set_trace() if self.use_amp: # An example self.model, optimizer = amp.initialize( 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