ohhhyeahhh / SiamGAT

Code for the paper "Graph Attention Tracking". (CVPR2021)
https://openaccess.thecvf.com/content/CVPR2021/papers/Guo_Graph_Attention_Tracking_CVPR_2021_paper.pdf
138 stars 19 forks source link

关于论文实验结果的复现问题。 #37

Open ahfywf opened 2 years ago

ahfywf commented 2 years ago

代码下载下来后在没有调整任何参数的情况下进行训练在GOT10K上复现的AO只有0.60。

我的设备是单卡的3090,而论文的平台是4卡的2080ti。请问有哪些因素可能会影响我复现的结果?

twotwo2 commented 2 years ago

你好,测试时cfg.TRACK.LR,cfg.TRACK.PENALTY_K,cfg.TRACK.WINDOW_INFLUENCE三个参数会对跟踪效果产生较大的影响。对于重新训练的模型,请问有尝试过微调这三个参数吗,或许能得到一些提升。另外,我们发现使用较小的batch size来训练模型会取得较好的结果。

hongtaofly commented 2 years ago

你好,测试时cfg.TRACK.LR,cfg.TRACK.PENALTY_K,cfg.TRACK.WINDOW_INFLUENCE三个参数会对跟踪效果产生较大的影响。对于重新训练的模型,请问有尝试过微调这三个参数吗,或许能得到一些提升。另外,我们发现使用较小的batch size来训练模型会取得较好的结果。

您好,想请问大概多大的batchsize比较合适,或者说batchsize的调整需要是2或4的倍数吗?

computerManAlan commented 2 years ago

想问问您单卡3090您的pytorch是什么版本呢,我训练到10epoch后就会报错

ahfywf commented 2 years ago

想问问您单卡3090您的pytorch是什么版本呢,我训练到10epoch后就会报错

我复现用的是pytorch11.1。 pytorch11.3也可以跑