size not match 600*600

[Cleehao]

only one side keep 640，and another side is changing😂

this is my output shape: torch.Size([1, 3, 928, 640]) #the first number is batchsize Epoch: 0 | Iteration: 67 | Cl: 1.22591 | Rl: 1.02222 | tl: 2.17842 | BT: 0.40437 s | ETA: 2 days, 8:16:40 torch.Size([1, 3, 640, 832]) Epoch: 0 | Iteration: 68 | Cl: 1.12365 | Rl: 1.10777 | tl: 2.17919 | BT: 0.39689 s | ETA: 2 days, 7:14:16 torch.Size([1, 3, 640, 832]) Epoch: 0 | Iteration: 69 | Cl: 1.22673 | Rl: 1.20870 | tl: 2.18285 | BT: 0.39252 s | ETA: 2 days, 6:37:42 torch.Size([1, 3, 928, 640])