Linfeng-Tang / SeAFusion

The code of " Image fusion in the loop of high-level vision tasks: A semantic-aware real-time infrared and visible image fusion network"
MIT License
184 stars 34 forks source link

关于训练 #38

Closed liushh39 closed 7 months ago

liushh39 commented 7 months ago

你好,训练一次分割器要80000次迭代,1个epoch得训练10个小时,我用的3070,是我的问题吗

Linfeng-Tang commented 7 months ago

是不是没有加载到GPU上面去呀。

liushh39 commented 7 months ago

哦哦,是4个epoch10小时

  发自我的企业微信         

----------回复的邮件信息---------- Linfeng @.***> 于2024年1月14日(周日) 中午11:52写道:

是不是没有加载到GPU上面去呀。

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