mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-26 13:00:58 +08:00
removed self.model refs
This commit is contained in:
1 parent
b58ec7ad5a
commit
a40b21bce0
1 file changed
+1
-1
@@ -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())
|
||||
|
||||
Reference in new issue
Block a user