DehazeNet_Pytorch icon indicating copy to clipboard operation
DehazeNet_Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon thuBingo

torch.nn里没有BReLU这个激活函数，要把作者自己写的BReLU添加到PyTorch源文件里

Open sun0912704 opened this issue 2 years ago • 2 comments

torch.nn里没有BReLU这个激活函数，要把作者自己写的BReLU添加到PyTorch源文件里

Originally posted by @Scotty1027 in https://github.com/thuBingo/DehazeNet_Pytorch/issues/1#issuecomment-723012141

sun0912704 avatar Mar 30 '22 06:03 sun0912704

怎么在源文件中添加呢？

sun0912704 avatar Mar 30 '22 06:03 sun0912704

怎么在源文件中添加呢？

我觉得把, 作者代码里面有点问题, 他在DehazeNet()这个类里面定义了BReLu()函数, 所以在forward()里面直接用self.BReLu()就行了, 不过就如作者所说的, 这个复现网络的去雾效果并不理想, 所以仅供参考

1074224619 avatar May 16 '22 08:05 1074224619