search

dansuh17 / segan-pytorch

SEGAN pytorch implementation https://arxiv.org/abs/1703.09452
GNU General Public License v3.0
106 stars 32 forks source link

Implement and use virtual batch norm instead of regular batch norm. #1

Closed dansuh17 closed 6 years ago

dansuh17 commented 6 years ago

https://discuss.pytorch.org/t/parameter-grad-of-conv-weight-is-none-after-virtual-batch-normalization/9036/6