From 76bc2fa75b5ce0a4503c258138f1562dfdf072c5 Mon Sep 17 00:00:00 2001 From: yoshikisd Date: Fri, 1 Aug 2025 21:50:35 +0000 Subject: [PATCH] Got rid of world_size attribute --- src/cdtools/reconstructors/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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