Train issue

[LeoMyMe]

Why warped_cloth and it's mask missing after 294000.

And there is also an another problem that I must change the train code.↓ Traceback (most recent call last): File "train_PBAFN_stage1.py", line 116, in loss_vgg = criterionVGG(x_all[num], cur_person_clothes.cuda()) File "C:\Users\Leo\anaconda3\envs\PMC2D\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "C:\Users\Leo\Desktop\PF-AFN\PF-AFN_train\models\networks.py", line 164, in forward x_vgg, y_vgg = self.vgg(x), self.vgg(y) File "C:\Users\Leo\anaconda3\envs\PMC2D\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "C:\Users\Leo\Desktop\PF-AFN\PF-AFN_train\models\networks.py", line 150, in forward h_relu5 = self.slice5(h_relu4) File "C:\Users\Leo\anaconda3\envs\PMC2D\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "C:\Users\Leo\anaconda3\envs\PMC2D\lib\site-packages\torch\nn\modules\container.py", line 204, in forward input = module(input) File "C:\Users\Leo\anaconda3\envs\PMC2D\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "C:\Users\Leo\anaconda3\envs\PMC2D\lib\site-packages\torch\nn\modules\pooling.py", line 166, in forward return F.max_pool2d(input, self.kernel_size, self.stride, File "C:\Users\Leo\anaconda3\envs\PMC2D\lib\site-packages\torch_jit_internal.py", line 485, in fn return if_false(*args, **kwargs) File "C:\Users\Leo\anaconda3\envs\PMC2D\lib\site-packages\torch\nn\functional.py", line 782, in _max_pool2d return torch.max_pool2d(input, kernel_size, stride, padding, dilation, ceil_mode) RuntimeError: Given input size: (512x2x1). Calculated output size: (512x1x0). Output size is too small

[LeoMyMe]

--niter 100 --niter_decay 100

[LeoMyMe]

--niter 100 --niter_decay 100

[Rockdow]

@LeoMyMe Hi, I'm facing the same problem. Have you found out the reason?