songdejia / Siamese-RPN-pytorch

This is a re-implementation of Siamese-RPN with pytorch, which is CVPR2018 spotlight.
MIT License
379 stars 87 forks source link

多batch训练 #14

Open BigPuns opened 5 years ago

BigPuns commented 5 years ago

这个训练的时候能多个batch来练嘛?每次取一个视频帧对来练,走完一个epoch都要好久。

xujiao3628 commented 5 years ago

这个训练的时候能多个batch来练嘛?每次取一个视频帧对来练,走完一个epoch都要好久。

我也想问这个,训练的时候刷刷的数一个一个往外蹦

BigPuns commented 5 years ago

你练的是什么数据?主动跟踪效果怎么样? @xujiao3628

xujiao3628 commented 5 years ago

你练的是什么数据?主动跟踪效果怎么样? @xujiao3628

我就试了一下

zzpustc commented 5 years ago

@BigPuns @xujiao3628 难道不能用大一点的batch训练么 我很奇怪好多个代码都是按照batch_size=1来训练的,如果训练数据是ILSVRC和youtube-bb的话 那得训练到什么时候……

wangaixue commented 5 years ago

他们训练肯定是 多卡 ,batch_size = 1要好久 更何况youtube这种数据 能提供一下youtube数据的分享吗?

woailuoshenqi commented 5 years ago

请问,为什么我训练的过程中会出现closs_ == nan ?

BigPuns commented 5 years ago

@woailuoshenqi 应该是你哪里搞错了,数据,代码,自己检查一下吧

KMSZJJ commented 5 years ago

@BigPuns 你好,我想请问一下data_loader.py中涉及的数据集TrainDataLoader(simple_vot13)在哪里下载?

BigPuns commented 5 years ago

@KMSZJJ 搜索一下啊vot2013的数据集,跟踪基本的评测集啊

klyjb commented 5 years ago

@BigPuns @xujiao3628 请问你们是如何解决多batch 训练的,谢谢大佬们。