Around line 330 in mm_interleaved.py, the shape of mmfs_features[i] is [B_I,1,C,H,W], where the 1st dimension of each mmfs_features[i] should be repeated to B_I to ensure that it will not go beyond the index when extracting the features of the corresponding image.