zhaodongxi
Results
1
comments of
zhaodongxi
> > 找到‘训练自己数据集map无法提高的原因了’,我提交的程序有个小问题,就是baseline训练(无稀疏化)时,默认打开了稀疏化,这可能是导致map无法提高的原因。解决办法是:在train文件的parser.add_argument中,将prune参数的默认值改为-1,当前的默认值为0(为0则打开了稀疏化) > > 改为-1之后,代码就跑不了了 if hasattr(model, 'module'): if opt.prune!=-1:BNOptimizer.updateBN(sr_flag, model.module.module_list, opt.s, prune_idx) else: if opt.prune!=-1:BNOptimizer.updateBN(sr_flag, model.module_list, opt.s, prune_idx) 不仅要改-1,这里也要改,改完可以跑