zkcys001 / UDAStrongBaseline

Open-source stronger baseline for unsupervised or domain adaptive object re-ID.
141 stars 21 forks source link

请问可以使用单卡训练吗?大概需要多少显存? #11

Closed LeeYushan closed 3 years ago

henrysun1996 commented 3 years ago

我试过了,所有的第二阶段的代码必须使用4块GPU跑。我用3块都跑不起来

zkcys001 commented 3 years ago

目前第二阶段是只支持4卡的,单卡性能会降低很多

mookixiao commented 10 months ago

目前第二阶段是只支持4卡的,单卡性能会降低很多

请问第二阶段为什么只支持四卡呢?理论上超参都设为相同精度结果应该是差不多的吧? 我现在正在用单张3090 24gb训练uncertainty,正在等结果