search

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

关于增加大循环次数出现报错 #20

Open JasonWong30 opened 1 year ago

JasonWong30 commented 1 year ago

您好。我尝试增加您这个模型代码Main函数下i的次数，但是每次当i=5的时候，总是报错如下图。请问这是什么原因？期待您的回复，谢谢！

image

2

Linfeng-Tang commented 1 year ago

momentum = 0.9 weight_decay = 5e-4 lr_start = 1e-2 max_iter = 80000 power = 0.9 warmup_steps = 1000 warmup_start_lr = 1e-5 it_start = i*20000 iter_nums=20000 这里面的对应的参数也需要修改的。

JasonWong30 commented 1 year ago

好的，我知道您的意思了。但是我还有个问题，就是您这里学习率计算是如此复杂，为什么不直接选用一个固定的学习率呢？

Linfeng-Tang commented 1 year ago

这是直接参考的BANet论文设置的学习率 最开始对语义分割任务了解的不是特别透彻 所以借鉴了一个现成的算法的代码

JasonWong30 commented 1 year ago

好的，非常感谢您耐心的回答，你真的太nice了！哈哈哈哈！

songwenhao123 commented 8 months ago

你好，我之前没有了解过分割，看了论文还是不太明白这些参数该如何调整，您可以告诉我该如何修改吗？谢谢！

Linfeng-Tang commented 8 months ago

max_iter = 80000 应该是主要修改max_iter，这个超参要根据你循环的次数i和每次循环里面迭代的次数进行调整。