search

Liling-yang / LENFsuion

The code of "LENFusion: A Joint Low-Light Enhancement and Fusion Network for Nighttime Infrared and Visible Image Fusion"
15 stars 2 forks source link

训练 #1

Open 123abxc opened 6 months ago

123abxc commented 6 months ago

能方便问下训练的设备内存吗 在训练时总是pycharm闪退内存不足

Liling-yang commented 6 months ago

train.py中的train_enhance() 是2438MiB，train_fusion() 是15420MiB； LFN_train.py是3672MiB； 所以完整训练整个网络结构显存至少要大于15420MiB

123abxc commented 6 months ago

好的谢谢您

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: Liling Yang @.> 发送时间: 2024年5月15日 16:15 收件人: Liling-yang/LENFsuion @.> 抄送: 123abxc @.>, Author @.> 主题: Re: [Liling-yang/LENFsuion] 训练 (Issue #1)

train.py中的train_enhance() 是2438MiB，train_fusion() 是15420MiB； LFN_train.py是3672MiB； 所以完整训练整个网络结构显存至少要大于15420MiB

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>