tried other img size? like [960,540]?

[yishuaidu]

I did that, but not worked..

[zhanghongyong123456]

I did that, but not worked..

mybe you need chage image size ，like w=540，h=540 ，

[yishuaidu]

does it work for you?

[yishuaidu]

I did that, but not worked..

mybe you need chage image size ，like w=540，h=540 ，

371083332 add me

[zhanghongyong123456]

I did that, but not worked..

mybe you need chage image size ，like w=540，h=540 ，

371083332 add me What is the error prompt?，if torch.cuda.OutOfMemoryError: CUDA out of memory ，you need set small batch size

[pHaeusler]

Yep. You need to change the w and h params