From cc3905cdc5dfc129f4d0b4e6995fd66c4fdb8d85 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 20:17:55 -0400 Subject: [PATCH] removed from_lightning flag --- pytorch_lightning/models/trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 9af6c394..4f999a71 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -461,7 +461,7 @@ class Trainer(TrainerIO): self.batch_loss_value = 0 # accumulated grads # init progbar when requested - if self.progress_bar : + if self.progress_bar: self.prog_bar = tqdm.tqdm(range(self.total_batches), position=self.process_position) for batch_nb, data_batch in enumerate(self.tng_dataloader): @@ -595,6 +595,7 @@ class Trainer(TrainerIO): # backward pass if self.use_amp: + # scale loss when using amp for optimizer in self.optimizers: with amp.scale_loss(loss, optimizer) as scaled_loss: scaled_loss.backward()