search

zcablii / SARDet_100K

Offical implementation of MSFA and release of SARDet_100K dataset for Large-Scale Synthetic Aperture Radar (SAR) Object Detection
Other
328 stars 25 forks source link

关于训练时长 #22

Closed cxq1 closed 2 months ago

cxq1 commented 2 months ago

你们的模型需要多久的训练时长，单卡A100能否训练，会不会很慢

zcablii commented 2 months ago

和模型有关。以VAN_base为例，imagenet预训练单卡估计要10天左右。如果是dota或者SARDet-100K上检测训练，估计两天。具体模型的具体时间可以根据我们提供的log推算。

cxq1 commented 2 months ago

谢谢，你的回复，另外我想问一下 单卡跑 学习率 应该怎么调整会比较合适一些

zcablii commented 2 months ago

可尝试根据batch size等比例缩放。比如batch size 16的lr为1e-4， 当使用batch size 8时，可尝试lr 5e-5。

cxq1 commented 2 months ago

可尝试根据batch size等比例缩放。比如batch size 16的lr为1e-4， 当使用batch size 8时，可尝试lr 5e-5。

谢谢，这里的batch size是指 多卡总的batch size嘛

zcablii commented 2 months ago

是的

cxq1 commented 2 months ago

是的

好的 感谢 我等下试试