关于训练遇到的问题

[rememberBr]

您好，我在尝试使用您的代码进行训练时发现，虽然论文以及github的readme里都是使用的SIRST数据集，但是在train_alcnet.py中数据集设置里默认值却是DENTIST和Iceberg，contrast.py里有对数据预处理的代码，也是针对这两个数据集的，我尝试自己修改但是遇到了其他的问题。 train_alcnet.py的参数设置里，选择的是PCMNet，但是只提供了MPCMResNetFPN和ResNetFCN的初始化代码，我尝试直接修改net_choice=‘MPCMResNetFPN’以及从model中导入PCMNet，但是代码都卡在了model.initialize(init=init.MSRAPrelu(), ctx=args.ctx, force_reinit=True)，请问该怎么办呢，能不能整理一份可执行的代码，非常感谢

[CPF-314127]

请问您解决了吗 我也遇到了这个问题

[CPF-314127]

请问您解决了吗 我也遇到了这个问题

[rememberBr]

并没有呢，他其实也不是卡住，是模型在初始化，需要等很久，我记得挂那里等了很久模型终于开始动了，但是损失不降，然后训练了200多轮的时候突然报错，我就放弃了....

[CPF-314127]

好的，谢谢您

[rememberBr]

你可以试试该作者的另一个模型，open-acm，好像比这个更新一点，但是我也没试过

[CPF-314127]

好的，非常感谢！