removed self.model refs

2026-09-09 11:32:07 +08:00 · 2019-06-26 18:27:25 -04:00
parent b58ec7ad5a
commit a40b21bce0
1 changed files with 1 additions and 1 deletions
@@ -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())