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

weight initialization #34

Open
evagian opened this issue May 23, 2018 · 4 comments
Open

weight initialization #34

evagian opened this issue May 23, 2018 · 4 comments

Comments

@evagian
Copy link

evagian commented May 23, 2018

Hello :)
Could you please help me with something?...
It is not clear to me in which part of your code you initialize the weights of the CNN.

@anlijuncn
Copy link

It's also confused to me too. I recommend you to read https://github.com/SaoYan/DnCNN-PyTorch
I don't know if we need some skills to initialize these weights with BN layers. If you have reached to the final result in paper. Please SHARE on you github!! I am now working on it, but only could get PSNR 29.14 for DnCNN-S. :)

@evagian
Copy link
Author

evagian commented Sep 27, 2018

Thank you @anlijuncn

I think that they are using the default tf.layers.conv2d initializer which is Xavier but I am not sure about that.

@anlijuncn
Copy link

Yes, in crisb-dut's code he is using default initializer. I don't know whether methods could improve the result. The author ZhangKai has released Pytorch and Keras version code on github/cszn.

@eastchun
Copy link

Default kernel initializer for tf.layer.conv2d is "glorot_uniform_initializer".

Please refer to code lines '315 - 317' in : https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/ops/variable_scope.py

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

3 participants