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

运行main.py出错 #2

Open
GewelsJI opened this issue Jun 29, 2019 · 7 comments
Open

运行main.py出错 #2

GewelsJI opened this issue Jun 29, 2019 · 7 comments

Comments

@GewelsJI
Copy link

RuntimeError: Error(s) in loading state_dict for DataParallel:
Unexpected key(s) in state_dict: "module.feature.features.norm0.num_batches_tracked", "module.feature.features.denseblock1.0.norm1.num_batches_tracked", "module.feature.features.denseblock1.0.norm2.num_batches_tracked", "module.feature.features.denseblock1.1.norm1.num_batches_tracked", "module.feature.features.denseblock1.1.norm2.num_batches_tracked", "module.feature.features.denseblock1.2.norm1.num_batches_tracked", "module.feature.features.denseblock1.2.norm2.num_batches_tracked", "module.feature.features.denseblock1.3.norm1.num_batches_tracked", "module.feature.features.denseblock1.3.norm2.num_batches_tracked", "module.feature.features.denseblock1.4.norm1.num_batches_tracked", "module.feature.features.denseblock1.4.norm2.num_batches_tracked", "module.feature.features.denseblock1.5.norm1.num_batches_tracked", "module.feature.features.denseblock1.5.norm2.num_batches_tracked", "module.feature.features.transition1.0.num_batches_tracked", "module.feature.features.denseblock2.0.norm1.num_batches_tracked", "module.feature.features.denseblock2.0.norm2.num_batches_tracked", "module.feature.features.denseblock2.1.norm1.num_batches_tracked", "module.feature.features.denseblock2.1.norm2.num_batches_tracked", "module.feature.features.denseblock2.2.norm1.num_batches_tracked", "module.feature.features.denseblock2.2.norm2.num_batches_tracked", "module.feature.features.denseblock2.3.norm1.num_batches_tracked", "module.feature.features.denseblock2.3.norm2.num_batches_tracked", "module.feature.features.denseblock2.4.norm1.num_batches_tracked", "module.feature.features.denseblock2.4.norm2.num_batches_tracked", "module.feature.features.denseblock2.5.norm1.num_batches_tracked", "module.feature.features.denseblock2.5.norm2.num_batches_tracked", "module.feature.features.denseblock2.6.norm1.num_batches_tracked", "module.feature.features.denseblock2.6.norm2.num_batches_tracked", "module.feature.features.denseblock2.7.norm1.num_batches_tracked", "module.feature.features.denseblock2.7.norm2.num_batches_tracked", "module.feature.features.denseblock2.8.norm1.num_batches_tracked", "module.feature.features.denseblock2.8.norm2.num_batches_tracked", "module.feature.features.denseblock2.9.norm1.num_batches_tracked", "module.feature.features.denseblock2.9.norm2.num_batches_tracked", "module.feature.features.denseblock2.10.norm1.num_batches_tracked", "module.feature.features.denseblock2.10.norm2.num_batches_tracked", "module.feature.features.denseblock2.11.norm1.num_batches_tracked", "module.feature.features.denseblock2.11.norm2.num_batches_tracked", "module.feature.features.transition2.0.num_batches_tracked", "module.feature.features.denseblock3.0.norm1.num_batches_tracked", "module.feature.features.denseblock3.0.norm2.num_batches_tracked", "module.feature.features.denseblock3.1.norm1.num_batches_tracked", "module.feature.features.denseblock3.1.norm2.num_batches_tracked", "module.feature.features.denseblock3.2.norm1.num_batches_tracked", "module.feature.features.denseblock3.2.norm2.num_batches_tracked", "module.feature.features.denseblock3.3.norm1.num_batches_tracked", "module.feature.features.denseblock3.3.norm2.num_batches_tracked", "module.feature.features.denseblock3.4.norm1.num_batches_tracked", "module.feature.features.denseblock3.4.norm2.num_batches_tracked", "module.feature.features.denseblock3.5.norm1.num_batches_tracked", "module.feature.features.denseblock3.5.norm2.num_batches_tracked", "module.feature.features.denseblock3.6.norm1.num_batches_tracked", "module.feature.features.denseblock3.6.norm2.num_batches_tracked", "module.feature.features.denseblock3.7.norm1.num_batches_tracked", "module.feature.features.denseblock3.7.norm2.num_batches_tracked", "module.feature.features.denseblock3.8.norm1.num_batches_tracked", "module.feature.features.denseblock3.8.norm2.num_batches_tracked", "module.feature.features.denseblock3.9.norm1.num_batches_tracked", "module.feature.features.denseblock3.9.norm2.num_batches_tracked", "module.feature.features.denseblock3.10.norm1.num_batches_tracked", "module.feature.features.denseblock3.10.norm2.num_batches_tracked", "module.feature.features.denseblock3.11.norm1.num_batches_tracked", "module.feature.features.denseblock3.11.norm2.num_batches_tracked", "module.feature.features.denseblock3.12.norm1.num_batches_tracked", "module.feature.features.denseblock3.12.norm2.num_batches_tracked", "module.feature.features.denseblock3.13.norm1.num_batches_tracked", "module.feature.features.denseblock3.13.norm2.num_batches_tracked", "module.feature.features.denseblock3.14.norm1.num_batches_tracked", "module.feature.features.denseblock3.14.norm2.num_batches_tracked", "module.feature.features.denseblock3.15.norm1.num_batches_tracked", "module.feature.features.denseblock3.15.norm2.num_batches_tracked", "module.feature.features.denseblock3.16.norm1.num_batches_tracked", "module.feature.features.denseblock3.16.norm2.num_batches_tracked", "module.feature.features.denseblock3.17.norm1.num_batches_tracked", "module.feature.features.denseblock3.17.norm2.num_batches_tracked", "module.feature.features.denseblock3.18.norm1.num_batches_tracked", "module.feature.features.denseblock3.18.norm2.num_batches_tracked", "module.feature.features.denseblock3.19.norm1.num_batches_tracked", "module.feature.features.denseblock3.19.norm2.num_batches_tracked", "module.feature.features.denseblock3.20.norm1.num_batches_tracked", "module.feature.features.denseblock3.20.norm2.num_batches_tracked", "module.feature.features.denseblock3.21.norm1.num_batches_tracked", "module.feature.features.denseblock3.21.norm2.num_batches_tracked", "module.feature.features.denseblock3.22.norm1.num_batches_tracked", "module.feature.features.denseblock3.22.norm2.num_batches_tracked", "module.feature.features.denseblock3.23.norm1.num_batches_tracked", "module.feature.features.denseblock3.23.norm2.num_batches_tracked", "module.feature.features.denseblock3.24.norm1.num_batches_tracked", "module.feature.features.denseblock3.24.norm2.num_batches_tracked", "module.feature.features.denseblock3.25.norm1.num_batches_tracked", "module.feature.features.denseblock3.25.norm2.num_batches_tracked", "module.feature.features.denseblock3.26.norm1.num_batches_tracked", "module.feature.features.denseblock3.26.norm2.num_batches_tracked", "module.feature.features.denseblock3.27.norm1.num_batches_tracked", "module.feature.features.denseblock3.27.norm2.num_batches_tracked", "module.feature.features.denseblock3.28.norm1.num_batches_tracked", "module.feature.features.denseblock3.28.norm2.num_batches_tracked", "module.feature.features.denseblock3.29.norm1.num_batches_tracked", "module.feature.features.denseblock3.29.norm2.num_batches_tracked", "module.feature.features.denseblock3.30.norm1.num_batches_tracked", "module.feature.features.denseblock3.30.norm2.num_batches_tracked", "module.feature.features.denseblock3.31.norm1.num_batches_tracked", "module.feature.features.denseblock3.31.norm2.num_batches_tracked", "module.feature.features.transition3.0.num_batches_tracked", "module.feature.features.denseblock4.0.norm1.num_batches_tracked", "module.feature.features.denseblock4.0.norm2.num_batches_tracked", "module.feature.features.denseblock4.1.norm1.num_batches_tracked", "module.feature.features.denseblock4.1.norm2.num_batches_tracked", "module.feature.features.denseblock4.2.norm1.num_batches_tracked", "module.feature.features.denseblock4.2.norm2.num_batches_tracked", "module.feature.features.denseblock4.3.norm1.num_batches_tracked", "module.feature.features.denseblock4.3.norm2.num_batches_tracked", "module.feature.features.denseblock4.4.norm1.num_batches_tracked", "module.feature.features.denseblock4.4.norm2.num_batches_tracked", "module.feature.features.denseblock4.5.norm1.num_batches_tracked", "module.feature.features.denseblock4.5.norm2.num_batches_tracked", "module.feature.features.denseblock4.6.norm1.num_batches_tracked", "module.feature.features.denseblock4.6.norm2.num_batches_tracked", "module.feature.features.denseblock4.7.norm1.num_batches_tracked", "module.feature.features.denseblock4.7.norm2.num_batches_tracked", "module.feature.features.denseblock4.8.norm1.num_batches_tracked", "module.feature.features.denseblock4.8.norm2.num_batches_tracked", "module.feature.features.denseblock4.9.norm1.num_batches_tracked", "module.feature.features.denseblock4.9.norm2.num_batches_tracked", "module.feature.features.denseblock4.10.norm1.num_batches_tracked", "module.feature.features.denseblock4.10.norm2.num_batches_tracked", "module.feature.features.denseblock4.11.norm1.num_batches_tracked", "module.feature.features.denseblock4.11.norm2.num_batches_tracked", "module.feature.features.denseblock4.12.norm1.num_batches_tracked", "module.feature.features.denseblock4.12.norm2.num_batches_tracked", "module.feature.features.denseblock4.13.norm1.num_batches_tracked", "module.feature.features.denseblock4.13.norm2.num_batches_tracked", "module.feature.features.denseblock4.14.norm1.num_batches_tracked", "module.feature.features.denseblock4.14.norm2.num_batches_tracked", "module.feature.features.denseblock4.15.norm1.num_batches_tracked", "module.feature.features.denseblock4.15.norm2.num_batches_tracked", "module.feature.features.denseblock4.16.norm1.num_batches_tracked", "module.feature.features.denseblock4.16.norm2.num_batches_tracked", "module.feature.features.denseblock4.17.norm1.num_batches_tracked", "module.feature.features.denseblock4.17.norm2.num_batches_tracked", "module.feature.features.denseblock4.18.norm1.num_batches_tracked", "module.feature.features.denseblock4.18.norm2.num_batches_tracked", "module.feature.features.denseblock4.19.norm1.num_batches_tracked", "module.feature.features.denseblock4.19.norm2.num_batches_tracked", "module.feature.features.denseblock4.20.norm1.num_batches_tracked", "module.feature.features.denseblock4.20.norm2.num_batches_tracked", "module.feature.features.denseblock4.21.norm1.num_batches_tracked", "module.feature.features.denseblock4.21.norm2.num_batches_tracked", "module.feature.features.denseblock4.22.norm1.num_batches_tracked", "module.feature.features.denseblock4.22.norm2.num_batches_tracked", "module.feature.features.denseblock4.23.norm1.num_batches_tracked", "module.feature.features.denseblock4.23.norm2.num_batches_tracked", "module.feature.features.denseblock4.24.norm1.num_batches_tracked", "module.feature.features.denseblock4.24.norm2.num_batches_tracked", "module.feature.features.denseblock4.25.norm1.num_batches_tracked", "module.feature.features.denseblock4.25.norm2.num_batches_tracked", "module.feature.features.denseblock4.26.norm1.num_batches_tracked", "module.feature.features.denseblock4.26.norm2.num_batches_tracked", "module.feature.features.denseblock4.27.norm1.num_batches_tracked", "module.feature.features.denseblock4.27.norm2.num_batches_tracked", "module.feature.features.denseblock4.28.norm1.num_batches_tracked", "module.feature.features.denseblock4.28.norm2.num_batches_tracked", "module.feature.features.denseblock4.29.norm1.num_batches_tracked", "module.feature.features.denseblock4.29.norm2.num_batches_tracked", "module.feature.features.denseblock4.30.norm1.num_batches_tracked", "module.feature.features.denseblock4.30.norm2.num_batches_tracked", "module.feature.features.denseblock4.31.norm1.num_batches_tracked", "module.feature.features.denseblock4.31.norm2.num_batches_tracked", "module.feature.features.denseblock4.32.num_batches_tracked", "module.feature.features.norm5.num_batches_tracked", "module.feature.features.block0.1.num_batches_tracked".

@GewelsJI
Copy link
Author

另外,您提供的pytorch的版本是0.5.0,我尚未在pytorch官网找到这个版本对应的pytorch

@zengxianyu
Copy link
Owner

RuntimeError: Error(s) in loading state_dict for DataParallel:
Unexpected key(s) in state_dict: "module.feature.features.norm0.num_batches_tracked", "module.feature.features.denseblock1.0.norm1.num_batches_tracked", "module.feature.features.denseblock1.0.norm2.num_batches_tracked", "module.feature.features.denseblock1.1.norm1.num_batches_tracked", "module.feature.features.denseblock1.1.norm2.num_batches_tracked", "module.feature.features.denseblock1.2.norm1.num_batches_tracked", "module.feature.features.denseblock1.2.norm2.num_batches_tracked", "module.feature.features.denseblock1.3.norm1.num_batches_tracked", "module.feature.features.denseblock1.3.norm2.num_batches_tracked", "module.feature.features.denseblock1.4.norm1.num_batches_tracked", "module.feature.features.denseblock1.4.norm2.num_batches_tracked", "module.feature.features.denseblock1.5.norm1.num_batches_tracked", "module.feature.features.denseblock1.5.norm2.num_batches_tracked", "module.feature.features.transition1.0.num_batches_tracked", "module.feature.features.denseblock2.0.norm1.num_batches_tracked", "module.feature.features.denseblock2.0.norm2.num_batches_tracked", "module.feature.features.denseblock2.1.norm1.num_batches_tracked", "module.feature.features.denseblock2.1.norm2.num_batches_tracked", "module.feature.features.denseblock2.2.norm1.num_batches_tracked", "module.feature.features.denseblock2.2.norm2.num_batches_tracked", "module.feature.features.denseblock2.3.norm1.num_batches_tracked", "module.feature.features.denseblock2.3.norm2.num_batches_tracked", "module.feature.features.denseblock2.4.norm1.num_batches_tracked", "module.feature.features.denseblock2.4.norm2.num_batches_tracked", "module.feature.features.denseblock2.5.norm1.num_batches_tracked", "module.feature.features.denseblock2.5.norm2.num_batches_tracked", "module.feature.features.denseblock2.6.norm1.num_batches_tracked", "module.feature.features.denseblock2.6.norm2.num_batches_tracked", "module.feature.features.denseblock2.7.norm1.num_batches_tracked", "module.feature.features.denseblock2.7.norm2.num_batches_tracked", "module.feature.features.denseblock2.8.norm1.num_batches_tracked", "module.feature.features.denseblock2.8.norm2.num_batches_tracked", "module.feature.features.denseblock2.9.norm1.num_batches_tracked", "module.feature.features.denseblock2.9.norm2.num_batches_tracked", "module.feature.features.denseblock2.10.norm1.num_batches_tracked", "module.feature.features.denseblock2.10.norm2.num_batches_tracked", "module.feature.features.denseblock2.11.norm1.num_batches_tracked", "module.feature.features.denseblock2.11.norm2.num_batches_tracked", "module.feature.features.transition2.0.num_batches_tracked", "module.feature.features.denseblock3.0.norm1.num_batches_tracked", "module.feature.features.denseblock3.0.norm2.num_batches_tracked", "module.feature.features.denseblock3.1.norm1.num_batches_tracked", "module.feature.features.denseblock3.1.norm2.num_batches_tracked", "module.feature.features.denseblock3.2.norm1.num_batches_tracked", "module.feature.features.denseblock3.2.norm2.num_batches_tracked", "module.feature.features.denseblock3.3.norm1.num_batches_tracked", "module.feature.features.denseblock3.3.norm2.num_batches_tracked", "module.feature.features.denseblock3.4.norm1.num_batches_tracked", "module.feature.features.denseblock3.4.norm2.num_batches_tracked", "module.feature.features.denseblock3.5.norm1.num_batches_tracked", "module.feature.features.denseblock3.5.norm2.num_batches_tracked", "module.feature.features.denseblock3.6.norm1.num_batches_tracked", "module.feature.features.denseblock3.6.norm2.num_batches_tracked", "module.feature.features.denseblock3.7.norm1.num_batches_tracked", "module.feature.features.denseblock3.7.norm2.num_batches_tracked", "module.feature.features.denseblock3.8.norm1.num_batches_tracked", "module.feature.features.denseblock3.8.norm2.num_batches_tracked", "module.feature.features.denseblock3.9.norm1.num_batches_tracked", "module.feature.features.denseblock3.9.norm2.num_batches_tracked", "module.feature.features.denseblock3.10.norm1.num_batches_tracked", "module.feature.features.denseblock3.10.norm2.num_batches_tracked", "module.feature.features.denseblock3.11.norm1.num_batches_tracked", "module.feature.features.denseblock3.11.norm2.num_batches_tracked", "module.feature.features.denseblock3.12.norm1.num_batches_tracked", "module.feature.features.denseblock3.12.norm2.num_batches_tracked", "module.feature.features.denseblock3.13.norm1.num_batches_tracked", "module.feature.features.denseblock3.13.norm2.num_batches_tracked", "module.feature.features.denseblock3.14.norm1.num_batches_tracked", "module.feature.features.denseblock3.14.norm2.num_batches_tracked", "module.feature.features.denseblock3.15.norm1.num_batches_tracked", "module.feature.features.denseblock3.15.norm2.num_batches_tracked", "module.feature.features.denseblock3.16.norm1.num_batches_tracked", "module.feature.features.denseblock3.16.norm2.num_batches_tracked", "module.feature.features.denseblock3.17.norm1.num_batches_tracked", "module.feature.features.denseblock3.17.norm2.num_batches_tracked", "module.feature.features.denseblock3.18.norm1.num_batches_tracked", "module.feature.features.denseblock3.18.norm2.num_batches_tracked", "module.feature.features.denseblock3.19.norm1.num_batches_tracked", "module.feature.features.denseblock3.19.norm2.num_batches_tracked", "module.feature.features.denseblock3.20.norm1.num_batches_tracked", "module.feature.features.denseblock3.20.norm2.num_batches_tracked", "module.feature.features.denseblock3.21.norm1.num_batches_tracked", "module.feature.features.denseblock3.21.norm2.num_batches_tracked", "module.feature.features.denseblock3.22.norm1.num_batches_tracked", "module.feature.features.denseblock3.22.norm2.num_batches_tracked", "module.feature.features.denseblock3.23.norm1.num_batches_tracked", "module.feature.features.denseblock3.23.norm2.num_batches_tracked", "module.feature.features.denseblock3.24.norm1.num_batches_tracked", "module.feature.features.denseblock3.24.norm2.num_batches_tracked", "module.feature.features.denseblock3.25.norm1.num_batches_tracked", "module.feature.features.denseblock3.25.norm2.num_batches_tracked", "module.feature.features.denseblock3.26.norm1.num_batches_tracked", "module.feature.features.denseblock3.26.norm2.num_batches_tracked", "module.feature.features.denseblock3.27.norm1.num_batches_tracked", "module.feature.features.denseblock3.27.norm2.num_batches_tracked", "module.feature.features.denseblock3.28.norm1.num_batches_tracked", "module.feature.features.denseblock3.28.norm2.num_batches_tracked", "module.feature.features.denseblock3.29.norm1.num_batches_tracked", "module.feature.features.denseblock3.29.norm2.num_batches_tracked", "module.feature.features.denseblock3.30.norm1.num_batches_tracked", "module.feature.features.denseblock3.30.norm2.num_batches_tracked", "module.feature.features.denseblock3.31.norm1.num_batches_tracked", "module.feature.features.denseblock3.31.norm2.num_batches_tracked", "module.feature.features.transition3.0.num_batches_tracked", "module.feature.features.denseblock4.0.norm1.num_batches_tracked", "module.feature.features.denseblock4.0.norm2.num_batches_tracked", "module.feature.features.denseblock4.1.norm1.num_batches_tracked", "module.feature.features.denseblock4.1.norm2.num_batches_tracked", "module.feature.features.denseblock4.2.norm1.num_batches_tracked", "module.feature.features.denseblock4.2.norm2.num_batches_tracked", "module.feature.features.denseblock4.3.norm1.num_batches_tracked", "module.feature.features.denseblock4.3.norm2.num_batches_tracked", "module.feature.features.denseblock4.4.norm1.num_batches_tracked", "module.feature.features.denseblock4.4.norm2.num_batches_tracked", "module.feature.features.denseblock4.5.norm1.num_batches_tracked", "module.feature.features.denseblock4.5.norm2.num_batches_tracked", "module.feature.features.denseblock4.6.norm1.num_batches_tracked", "module.feature.features.denseblock4.6.norm2.num_batches_tracked", "module.feature.features.denseblock4.7.norm1.num_batches_tracked", "module.feature.features.denseblock4.7.norm2.num_batches_tracked", "module.feature.features.denseblock4.8.norm1.num_batches_tracked", "module.feature.features.denseblock4.8.norm2.num_batches_tracked", "module.feature.features.denseblock4.9.norm1.num_batches_tracked", "module.feature.features.denseblock4.9.norm2.num_batches_tracked", "module.feature.features.denseblock4.10.norm1.num_batches_tracked", "module.feature.features.denseblock4.10.norm2.num_batches_tracked", "module.feature.features.denseblock4.11.norm1.num_batches_tracked", "module.feature.features.denseblock4.11.norm2.num_batches_tracked", "module.feature.features.denseblock4.12.norm1.num_batches_tracked", "module.feature.features.denseblock4.12.norm2.num_batches_tracked", "module.feature.features.denseblock4.13.norm1.num_batches_tracked", "module.feature.features.denseblock4.13.norm2.num_batches_tracked", "module.feature.features.denseblock4.14.norm1.num_batches_tracked", "module.feature.features.denseblock4.14.norm2.num_batches_tracked", "module.feature.features.denseblock4.15.norm1.num_batches_tracked", "module.feature.features.denseblock4.15.norm2.num_batches_tracked", "module.feature.features.denseblock4.16.norm1.num_batches_tracked", "module.feature.features.denseblock4.16.norm2.num_batches_tracked", "module.feature.features.denseblock4.17.norm1.num_batches_tracked", "module.feature.features.denseblock4.17.norm2.num_batches_tracked", "module.feature.features.denseblock4.18.norm1.num_batches_tracked", "module.feature.features.denseblock4.18.norm2.num_batches_tracked", "module.feature.features.denseblock4.19.norm1.num_batches_tracked", "module.feature.features.denseblock4.19.norm2.num_batches_tracked", "module.feature.features.denseblock4.20.norm1.num_batches_tracked", "module.feature.features.denseblock4.20.norm2.num_batches_tracked", "module.feature.features.denseblock4.21.norm1.num_batches_tracked", "module.feature.features.denseblock4.21.norm2.num_batches_tracked", "module.feature.features.denseblock4.22.norm1.num_batches_tracked", "module.feature.features.denseblock4.22.norm2.num_batches_tracked", "module.feature.features.denseblock4.23.norm1.num_batches_tracked", "module.feature.features.denseblock4.23.norm2.num_batches_tracked", "module.feature.features.denseblock4.24.norm1.num_batches_tracked", "module.feature.features.denseblock4.24.norm2.num_batches_tracked", "module.feature.features.denseblock4.25.norm1.num_batches_tracked", "module.feature.features.denseblock4.25.norm2.num_batches_tracked", "module.feature.features.denseblock4.26.norm1.num_batches_tracked", "module.feature.features.denseblock4.26.norm2.num_batches_tracked", "module.feature.features.denseblock4.27.norm1.num_batches_tracked", "module.feature.features.denseblock4.27.norm2.num_batches_tracked", "module.feature.features.denseblock4.28.norm1.num_batches_tracked", "module.feature.features.denseblock4.28.norm2.num_batches_tracked", "module.feature.features.denseblock4.29.norm1.num_batches_tracked", "module.feature.features.denseblock4.29.norm2.num_batches_tracked", "module.feature.features.denseblock4.30.norm1.num_batches_tracked", "module.feature.features.denseblock4.30.norm2.num_batches_tracked", "module.feature.features.denseblock4.31.norm1.num_batches_tracked", "module.feature.features.denseblock4.31.norm2.num_batches_tracked", "module.feature.features.denseblock4.32.num_batches_tracked", "module.feature.features.norm5.num_batches_tracked", "module.feature.features.block0.1.num_batches_tracked".

用0.4.1版的应该可以

@GewelsJI
Copy link
Author

GewelsJI commented Jul 4, 2019

依旧会出错,请问一下,我查了资料,是因为多卡GPU训练保存的权值文件,不能在单卡上面test,请问一下您以前是的哪几张卡?

@zengxianyu
Copy link
Owner

依旧会出错,请问一下,我查了资料,是因为多卡GPU训练保存的权值文件,不能在单卡上面test,请问一下您以前是的哪几张卡?

我用的0和1

@Elaineok
Copy link

依旧会出错,请问一下,我查了资料,是因为多卡GPU训练保存的权值文件,不能在单卡上面test,请问一下您以前是的哪几张卡?

你好,我也遇到相同问题,麻烦问下这个问题你解决了吗?是因为多卡问题?还是pytorch版本问题造成的错误呢?谢谢~

@GewelsJI
Copy link
Author

没有解决

@panmyuan
Copy link

依旧会出错,请问一下,我查了资料,是因为多卡GPU训练保存的权值文件,不能在单卡上面test,请问一下您以前是的哪几张卡?

我用的0和1

应该是保存模型是出错了,加个module就行了。

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

4 participants