search

MingTian99 / ESDNet

Learning A Spiking Neural Network for Efficient Image Deraining (IJCAI 2024)
39 stars 4 forks source link

自己訓練的成效不彰 #3

Open bee3131263 opened 3 months ago

bee3131263 commented 3 months ago

作者您好：

由於您提供的預訓練檔與模型對不上（如 Issue1），我自己將Rain200L丟入訓練，大約訓練了800個epoch，得出來Train_PSNR: 23.9329；SSIM: 0.6986。想請問我是否有遺漏了哪些細節呢？或者我是否應該對哪些超參數進行調整以獲得更好的功效？

非常感謝您的文章 希望能獲得您的回覆

zhaofengnain commented 3 months ago

请问你的参数设置是什么

zhaofengnain commented 3 months ago

@bee3131263

MingTian99 commented 3 months ago

感谢您的关注。关于训练的具体参数均已在train.py文件中进行了详细设置。请检查spikingjelly和cuda等版本是否一致。

bee3131263 commented 3 months ago

作者您好：

後來我將環境整個重建之後再次訓練，獲得出來的PSNR和SSIM皆與論文相近，造成上次的原因我仍不清楚。

已經沒有疑問了 非常感謝