search

mukulkhanna / FHDR

PyTorch Implementation of Global SIP 2019 Paper "FHDR: HDR Image Reconstruction from a SingleLDR Image using Feedback Network"
https://arxiv.org/abs/1912.11463
BSD 3-Clause "New" or "Revised" License
84 stars 18 forks source link

RuntimeError: normal_ expects std > 0.0, but found std=0 #2

Closed seyranakdemir closed 4 years ago

seyranakdemir commented 4 years ago

start_epoch = 1 model.apply(weights_init)

m.weight.data.normal(0.0, 0.0) I take error this code in train.py file=> RuntimeError: normal expects std > 0.0, but found std=0

Can you tell me, how can i get

mukulkhanna commented 4 years ago

You can try a very small normal value, like 1e-3, instead of 0.0.

m.weight.data.normal_(0.0, 1e-3)