removed self.model refs

2019-06-26 18:27:25 -04:00 · 2019-06-26 18:27:25 -04:00 · a40b21bce0
parent b58ec7ad5a
commit a40b21bce0
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -27,6 +27,7 @@ def reduce_distributed_output(output, nb_gpus):
            output[k] = reduced
    return output

+
 class Trainer(TrainerIO):

    def __init__(self,
@ -427,7 +428,6 @@ class Trainer(TrainerIO):
            loss.backward()

        if self.check_grad_nans:
-
            model = self.model.module if self.data_parallel else self.model
            for param in model.parameters():
                print(param.grad.float().sum())