tutorials icon indicating copy to clipboard operation
tutorials copied to clipboard

contrastive loss batch size

Open wyli opened this issue 3 years ago • 1 comments

Figured out a solution. In this line, the argument parsed into the definition of contrastive loss have 'batch_size=batch_size*2'. However, in the function, the batch size has been multiplied again. The problem was fixed when I set 'batch_size=batch_size' in the definition. If it is the case, I would send a PR to fix that.

Originally posted by @Eason-Qin in https://github.com/Project-MONAI/tutorials/discussions/624#discussioncomment-3389489

wyli avatar Aug 13 '22 14:08 wyli

this looks like a bug, @finalelement could you please help confirm?

wyli avatar Aug 13 '22 14:08 wyli

addressed by https://github.com/Project-MONAI/MONAI/pull/5093

wyli avatar Sep 14 '22 07:09 wyli