Remove line overwriting output (#2839)

jmarintur · jmarin · svekars · web-flow · commit 4c6033189de4 · 2024-04-24T10:52:47.000-07:00
Co-authored-by: jmarin &lt;javier.marin@satellogic.com&gt;
Co-authored-by: Svetlana Karslioglu &lt;svekars@meta.com&gt;
diff --git 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):
