Remove line overwriting output (#2839)

Co-authored-by: jmarin <javier.marin@satellogic.com>
Co-authored-by: Svetlana Karslioglu <svekars@meta.com>

Author: 3 people

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):