From 726dd1f61a493f4e896efa86942b6450439531b0 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 17:40:23 -0400 Subject: [PATCH] moved dataloaders after amp and optimizers --- pytorch_lightning/models/trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index bd0ed1119c..5b3be252fb 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -401,11 +401,11 @@ class Trainer(TrainerIO): self.lr_schedulers.append(scheduler) # print model summary - model.summarize() + ref_model.summarize() # give model convenience properties - model.trainer = self - model.experiment = self.experiment + ref_model.trainer = self + ref_model.experiment = self.experiment # run tiny validation to make sure program won't crash during val _ = self.validate(model, self.val_dataloader, max_batches=self.nb_sanity_val_steps)