thank you so much for pointing that out! I’m a newbie for python and pytorch and my dataset is too large that I didn’t detect such errors.
But I’m still struggling with this because I couldn’t exclude those data with different size otherwise the results will probably be biased…