Hello, thanks for your interesting research! https://github.com/VisionLearningGroup/OP_Match/blob/fd55de2e1913ffbcb3fd413bd9f024afce44c024/utils/misc.py#L67 On line 67, I think `unknown_size = len(ind)` should be replaced by `unknown_size = ind.sum()`. `len(ind)` is the batch size. Thanks for your help in advance.