Got rid of world_size attribute

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