diff --git a/src/cdtools/reconstructors/base.py b/src/cdtools/reconstructors/base.py
index 65b2771..d5d34a6 100644
--- a/src/cdtools/reconstructors/base.py
+++ b/src/cdtools/reconstructors/base.py
@@ -178,7 +178,7 @@ class Reconstructor:
                     loss.backward()
 
                     # Normalize the accumulating total loss
-                    total_loss += loss.detach() // self.model.world_size
+                    total_loss += loss.detach() 
 
                 # If we have a regularizer, we can calculate it separately,
                 # and the gradients will add to the minibatch gradient