ecnet's People
ecnet's Issues
Hello, can I ask you a few questions?
Thank you very much for sharing!
I'm sorry to bother you, because my foundation is really not very good, and I encountered several problems when running this code. I would really appreciate it if you could give me some guidance!
Fistly, in the code of "python train.py --dataroot./data/SPA-Data_6385_patch --name < autoencoder_path> --model autoencoder_train --dataset_mode rain100h --preprocess none --n_epochs 100 --lr_policy multistep Where gradient_clipping 5", does "autoencoder_path" refer to the path of the "autoencoder_train_model.py" file?
Secondly, if I'm right about the first question, I revised "autoencoder_path" to the path of the "autoencoder_train_model.py" file. Then I encountered this error: ModuleNotFoundError: No module named 'models. deraining. derainlcn_model'.
In the deraining folder, I do not find the file "derainlcn_model". Is it "RLCN.py"?
Wish you good health and smooth work during the epidemic! Looking forward to your reply!
About the training process
Hello, when I trained the rain100L dataset, the psnr value and ssim value have always been small, what is the reason why it does not match the data in the paper?
training
Hello,I see the training frameworks of visualizer_training and train are different. Can these two codes be used at the same time? If I use pycharm training it, the command should be same as the readme showing?
Pre-trained autoencoder loading error
Training ECNet would raise an error when tries loading the pre-trained autoencoder:
Unexpected key(s) in state_dict: "Up4.up.1.weight", "Up4.up.1.bias", "Up4.up.2.weight", "Up4.up.2.bias".....
It seems that the module is saved in DataParallel
when trains with GPU, which stores the model in module
, and is trying to load it without DataParallel.
# code snippet from build_networks.py
def init_net(net, init_type='normal', init_gain=0.02, gpu_ids=[], init=True):
if len(gpu_ids) > 0:
...
net = torch.nn.DataParallel(net, gpu_ids) # wrapped by DataParallel
# code snippet from base_model.py
def load_networks(self):
...
if isinstance(net, torch.nn.DataParallel):
net = net.module # unwrapped as nn.Module
state_dict = torch.load(load_path, map_location=str(self.device)) # the pre-trained network, save with DataParallel
net.load_state_dict(model_dict) # error happens here
RLCN
I want to know where have used RLCN in the code?I want to receive you reply! Thank you very much!
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.