【论文复现】UGATIT相关
Created by: CHFsky
paddle已有API:adaptive_pool2d, reshape, ones_like,zeros_like
unsqueeze,concat, create_parameter,L1Loss,MSELoss
sqrt, expand,reduce_sum,reduce_mean, transpose
自行定义API:ReflectionPad2D, ReLU, LeakyReLU,Tanh,
Upsample,,var, spectral_norm,BCEWithLogitsLoss,RhoClipper
所有API均通过随机数据进行了参数对齐测试,复现过程中谨慎地替换掉所有API名称并保证传参一致,
但是训练的Dloss降到0.0几,Gloss一直高于1400,无法继续下降,不知道问题出在哪,
因为程序跑通了,而且除了conv和linear无法直接对齐外,其余每一个api都进行了参数对齐测试,
所以想问,到底问题会出现在哪,一开始我猜测是网络参数对齐问题,但是测试没找出来,后来又觉得是损失部分问题,但是损失函数定义也是对齐的,过程也是一致的,
优化器的list是parameter()直接相加,先backword(),再minimize(),再optim.clear_gradients
当不进行预处理transforms,图片传入参数也是对齐的,实在找不到还有可能问题出在哪里,平台版本一直是1.8.3运行。
项目地址:
https://aistudio.baidu.com/bdvgpu/user/288399/759441/notebooks/759441.ipynb
因为这个问题卡了一周了,仍然无法解决,很多人是因为版本问题,切换后就正常了,我这个一直如此,希望能给个指点,因为损失不正常就终止了运行