FSRNET_pytorch
FSRNET_pytorch copied to clipboard
预训练模型前向传播报错
手动加载了weights.pkl后发现模型参数共有718个,但是define_G()新建的模型有716个参数,最后print两个模型对比得出,预训练的模型在prior_Estimation_Network的最后加了一层1x1的卷积层,models的代码里未作出改动。 而且预训练模型前向传播报错Upsample object has no attribute name,但是直接用define_G生成的模型前向传播不会报错,这个问题有点费解呀。
而且看了一眼输入图像是 0-255的uint8数据,预处理只有一个转tensor操作,求问输入还经过什么操作了吗
have you resolved the problem?