Xiaoli

@ryunchang I also got the same error with you, wondering if you have fixed the error, thanks

@ryunchang do you mean you modified line 182 in train.py `img = cv2.resize(img, (800, 288))` to 'img = cv2.resize(img, (296, 296))'?

I also tried both dim=0 and dim=1, both not working. @klintan would like to know any updates on this? thanks.

@mummy2358 Thank you. Will give it a try.

@mummy2358 I do not know why, on my side, after changing to `embedding_i = embedding_b * seg_mask_i` I still have the same problem. Not sure if it's computer related, as...