@想请教一下为什么我用这个代码进行训练之后再进行测试的时候结果很不好呢？甚至可以说掩码就基本没有处理，这是什么原因造成的呢？我用的数据集是Celeba-HQ,由于电脑配置的原因，训练周期为40个epoch，batch size 设置为4了，虽然源码的batch size为14，但我认为这个的影响不是很大，想请教一下是否是其他原因造成的。 #31 
