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

hello #10

Closed ziyunxuanzzz closed 5 months ago

ziyunxuanzzz commented 5 months ago

Is 12GB of memory sufficient for training this model with batch-size 1?

zcablii commented 5 months ago

Yes. If you're using a larger model and running out of GPU memory, you can try using a smaller image input size