RuntimeError: stack expects each tensor to be equal size, but got [1, 191, 512, 512] at entry 0 and [1, 181, 512, 512] at entry 1

[hilab-git]

RuntimeError: stack expects each tensor to be equal size, but got [1, 191, 512, 512] at entry 0 and [1, 181, 512, 512] at entry 1

[BJQ123456]

Hello, how did you solve this problem? thank you

[abcxubu]

I solved the problem. I did the RandomCrop to make all the cases have the same size, e.g., [96, 96, 96]. I hope it helps you.

[BJQ123456]

Thank you, I found the problem and solved it. I appreciate your response.