DeepFillv2_Pytorch icon indicating copy to clipboard operation
DeepFillv2_Pytorch copied to clipboard

无法训练

Open Dominic-ZZ opened this issue 4 years ago • 0 comments

几个epoch之后就爆显存,把batch_size设置为1或2时不会出现这个问题,但是在test时直接把输入图像输出了,mask部分变成了黑色

Dominic-ZZ avatar Jun 11 '21 02:06 Dominic-ZZ