From ecd0510536ae37a21b409c360e12ce6c05295f91 Mon Sep 17 00:00:00 2001 From: jmarin Date: Wed, 17 Apr 2024 20:27:45 +0000 Subject: [PATCH] Remove line overwritting output --- intermediate_source/per_sample_grads.py | 1 - 1 file changed, 1 deletion(-) diff --git a/intermediate_source/per_sample_grads.py b/intermediate_source/per_sample_grads.py index c423679229c..ece80d3f94f 100644 --- a/intermediate_source/per_sample_grads.py +++ b/intermediate_source/per_sample_grads.py @@ -42,7 +42,6 @@ def forward(self, x): x = F.relu(x) x = self.fc2(x) output = F.log_softmax(x, dim=1) - output = x return output def loss_fn(predictions, targets):