FSRNET_pytorch icon indicating copy to clipboard operation
FSRNET_pytorch copied to clipboard

预训练模型前向传播报错

Open little-one opened this issue 6 years ago • 2 comments

手动加载了weights.pkl后发现模型参数共有718个,但是define_G()新建的模型有716个参数,最后print两个模型对比得出,预训练的模型在prior_Estimation_Network的最后加了一层1x1的卷积层,models的代码里未作出改动。 而且预训练模型前向传播报错Upsample object has no attribute name,但是直接用define_G生成的模型前向传播不会报错,这个问题有点费解呀。

little-one avatar Dec 09 '19 08:12 little-one

而且看了一眼输入图像是 0-255的uint8数据,预处理只有一个转tensor操作,求问输入还经过什么操作了吗

little-one avatar Dec 10 '19 01:12 little-one

have you resolved the problem?

suke27 avatar Jan 15 '20 01:01 suke27