search

XPixelGroup / DepictQA

DepictQA: Depicted Image Quality Assessment with Vision Language Models
Apache License 2.0
68 stars 5 forks source link

训练和推理都只支持单卡吗？ #13

Closed pooregg closed 6 hours ago

pooregg commented 1 month ago

作者您好，请问训练和推理都只支持单张GPU吗，能否告知GPU型号，数量，训练时长？

zhiyuanyou commented 1 month ago

你好，

训练支持多卡，例如8卡训练：sh train.sh 0,1,2,3,4,5,6,7。

在8张A6000显卡上的训练时间约为20小时。