Skip to content

Commit fcd445c

Browse files
make fix
1 parent 12c3a3c commit fcd445c

File tree

1 file changed

+1
-1
lines changed
  • ssseg/modules/models/segmentors/mcibi

1 file changed

+1
-1
lines changed

ssseg/modules/models/segmentors/mcibi/mcibi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,4 +142,4 @@ def calculatememoryloss(self, stored_memory):
142142
loss_memory_log = loss_memory.data.clone()
143143
if dist.is_available() and dist.is_initialized():
144144
dist.all_reduce(loss_memory_log.div_(dist.get_world_size()), op=dist.ReduceOp.SUM)
145-
return loss_memory, loss_memory_log
145+
return loss_memory, loss_memory_log.item()

0 commit comments

Comments
 (0)