search

bingykang / Fewshot_Detection

Few-shot Object Detection via Feature Reweighting
https://arxiv.org/abs/1812.01866
526 stars 111 forks source link

Memory to train #12

Open duynn912 opened 4 years ago

duynn912 commented 4 years ago

Could you please let me know how much memory your model needs for training? My GPU has 12GB but I fail to start training though I put nw = 0 or 1 and down batchsize of yolo to 16 or 8?