search

hli1221 / imagefusion-rfn-nest

RFN-Nest(Information Fusion, 2021, Highly Cited Paper) - PyTorch =1.5，Python=3.7
108 stars 18 forks source link

关于模型训练，我们发现在第二轮训练时会出现模型不在cuda的问题 #9

Closed Zhou-wy closed 9 months ago

Zhou-wy commented 1 year ago

在train_fusionnet.py中第211-232行缩进是否有问题？ 按照我的理解，如果按照公布的源码训练，在第一轮训练后模型移往CPU，在第二轮训练开始后，模型在CPU 上，无法训练。应该把rain_fusionnet.py中第211-232行往前缩进吗？

hli1221 commented 1 year ago

谢谢你的反馈。 很抱歉，这确实是我的失误，最终保存模型的代码在整理程序时缩进太多。 我已经重新修改，并提交了。