YuanWu

Nanjing University of Science and Technology Nanjing, China PhD student at PCALab, NJUST, China

Results 3 comments of


                                            YuanWu

Can you provide the code for training? Thanks

@tkone2018 你好，你可以在作者 predict.py 文件下 ”if args.mode == 'demo':“ 语句后面加上 ”with torch.no_grad ():“。我加上之后程序可以运行，之前也是 out of memory

> 我是我根据论文自己复现的基于pytorch的代码 https://github.com/7568/DeRaindrop ，里面有训练代码和测试代码

@7568 您好，您发的链接好像打不开了，可以再重新发一下吗？谢谢！

The Global Average Pooling

To avoid error when batchsize = 1