PeterH0323/Smart_Construction

训练设备

xwq-jfl opened this issue · 1 comments

xwq-jfl commented

请问训练的时候怎么指定gpu设备

PeterH0323 commented

通过 --device 参数进行 GPU 的选择