Can you provide the code for training? Thanks

[Jay-Jia]

No description provided.

[world2025]

你好，可以分享下数据集吗？

[world2025]

我看了你们的报告，很感兴趣，谢谢！

[Ir1d]

@tkone2018 https://github.com/rui1996/DeRaindrop/blob/master/README.md#dataset

[world2025]

@Ir1d 好的，没注意

[world2025]

@Ir1d do you realize the demo ?

[Ir1d]

@tkone2018 demo 没问题啊？ training code 的确没有

[world2025]

@Ir1d 你理解那个attention模块吗？

[world2025]

@Ir1d 我用demo的时候显示cuda内存不够，很尴尬

[YiLiangNie]

When is it possible to provide training code for the algorithm? I have tested some pictures from web, and found that the effect is not good.

[MaybeShewill-CV]

@YiLiangNie Welcome to try my implementation and train your own model:) https://github.com/MaybeShewill-CV/attentive-gan-derainnet

[7568]

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

[zrt791521360]

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

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

你好，我用你的代码进行训练，训练结果和原文给出的权重差一点。这和您没有使用dataloader、和没有使用动态学习率有关吗

[zrt791521360]

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

@7568 你好，我用你的代码进行训练，训练结果和原文给出的权重差一点。这和您没有使用dataloader、和没有使用动态学习率有关吗

[Rayn-Wu]

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

[world2025]

@Rayn-Wu 好的，谢谢

[MarikaArimoto]

@7568
Hello.
I would also like to try your Pytorch implementation, but your page seems to be closed.
( https://github.com/7568/DeRaindrop )
Could you please share it again?
Thanks.