search

fh2019ustc / PolySnake

The official code for “Recurrent Generic Contour-based Instance Segmentation with Progressive Learning”, TCSVT, 2024.
MIT License
69 stars 7 forks source link

COCO训练时间 #12

Closed ZitengXue closed 8 months ago

ZitengXue commented 8 months ago

老哥您好，我最近复现了您的代码，但是有一些不解，我的设备是双卡4090 1.训练coco batch size 为32， 1个epoch训练时间近两个小时，请问这个训练速度正常吗？感觉如果按照默认config双卡4090需要150~300小时。 2.训练KINS batch size为16 显存会不够，只能batch size为8进行训练，想问下老哥当时KINS是用什么显卡配置跑的 谢谢老哥

fh2019ustc commented 8 months ago
  1. 正常的，COCO比较大，用3090会比2080快不少。你可以把卡数提到4，即4卡3090，把显存拉满
  2. KINS我是用2080跑的，12g显存
ZitengXue commented 8 months ago

非常感谢老哥！