Merge branch 'torch' of https://github.com/apax-hub/apax into torch

apax-hub · Apr 8, 2024 · 7fe06c1 · 7fe06c1
2 parents 0552d4d + 9f3f3c0
commit 7fe06c1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/apax/nn/torch/model/gmnn.py b/apax/nn/torch/model/gmnn.py
@@ -137,14 +137,14 @@ def forward(
             requires_grad.append(eps)
         else:
             perturbation = None
-        
+
         energy = self.energy_model(R, Z, neighbor, box, offsets, perturbation)
-                    
+
 
         grads = autograd.grad(energy, requires_grad,
                             grad_outputs=torch.ones_like(energy),
                             create_graph=True)
-        
+
         neg_forces = grads[0]
         forces = -neg_forces