We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 12c3a3c commit fcd445cCopy full SHA for fcd445c
ssseg/modules/models/segmentors/mcibi/mcibi.py
@@ -142,4 +142,4 @@ def calculatememoryloss(self, stored_memory):
142
loss_memory_log = loss_memory.data.clone()
143
if dist.is_available() and dist.is_initialized():
144
dist.all_reduce(loss_memory_log.div_(dist.get_world_size()), op=dist.ReduceOp.SUM)
145
- return loss_memory, loss_memory_log
+ return loss_memory, loss_memory_log.item()
0 commit comments