Comments (6)
你好,我也遇到了相似的错误,RuntimeError: Error(s) in loading state_dict for UnetGenerator:
我搜索到的解决方法是将base_model.py中的network.load_state_dict(torch.load(save_path))改为network.load_state_dict(torch.load(save_path),False)就可以完成模型加载了,希望能帮到你 @yumouse
from enlightengan.
@wasdyouli thanks a lot first~ Could I ask all your modifications based on the latest master branch code to test models with CPU? I'm not sure whether some modifications made before is valid or redundant and whether they would conflict with yours.
from enlightengan.
@yumouse 抱歉,我也是刚好遇到和你相似的问题,目前还没有完成在CPU上的测试,你可以先尝试我说的方法,根据你之前的描述,模型加载部分应该是可以完成的 .
from enlightengan.
@wasdyouli Ok, thx anyway! I will keep trying.
from enlightengan.
@yumouse 请问一下python scripts/script.py --predict或者 python predict.py,是否需要翻墙用外网?
from enlightengan.
@yumouse no need
from enlightengan.
Related Issues (20)
- Which model should be chose? HOT 1
- exchange trainA and trainB HOT 5
- why does image_numpy need to add one,and divide 2,finally multipy 255 HOT 2
- Code for calculate NIQE HOT 1
- Problems about how to fine-tuning HOT 3
- why loss is nan
- predict.py: error: the following arguments are required: --dataroot HOT 2
- UserWarning: nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.
- multiprocessing error HOT 2
- Could it be posible create low-light image? HOT 1
- 请问有关于纯粹的EnlightenGAN这篇论文的代码吗?不包含其他对比算法的。 HOT 1
- train issues HOT 1
- where to place the final_dataset folder? HOT 1
- 请问在哪里修改生成图像的名称 HOT 1
- error HOT 1
- TypeError: load() missing 1 required positional argument: 'Loader' HOT 1
- The difference between D_A and D_P loss
- 训练时判别器网络报错
- 下载模型 HOT 1
- Pre-Processing for Improving Classification
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from enlightengan.