您好,当我将QAM加入到STCN当中时,且输入到Score模块的特征是fuse后的,但是训练得到的score值到后面都是0了(Score模块里面全连接层的激活函数一个是relu,一个是sigmoid)我是1GPU训练,batch size为16,学习率为1e-5
lzl2040 opened this issue · 7 comments
lzl2040 commented
您好,当我将QAM加入到STCN当中时,且输入到Score模块的特征是fuse后的,但是训练得到的score值到后面都是0了(Score模块里面全连接层的激活函数一个是relu,一个是sigmoid)我是1GPU训练,batch size为16,学习率为1e-5
yongliu20 commented
都用sigmoid试一下?原版QAM使用的都是sigmoid。也可以检查一下你加到STCN里时gt的iou等处是否存在bug
1359347500cwc commented
TianXuXiang commented
1359347500cwc commented
TianXuXiang commented
lzl2040 commented
我之前跑的时候就是用sigmoid的
MrHsj commented