Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

预训练模型前向传播报错 #15

Open
little-one opened this issue Dec 9, 2019 · 2 comments
Open

预训练模型前向传播报错 #15

little-one opened this issue Dec 9, 2019 · 2 comments

Comments

@little-one
Copy link

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

@little-one
Copy link
Author

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

@suke27
Copy link

suke27 commented Jan 15, 2020

have you resolved the problem?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants