yolox-pytorch icon indicating copy to clipboard operation
yolox-pytorch copied to clipboard
verified publisher icon zhangming8

训练时 cuda 指定问题

Open jolt2017 opened this issue 4 years ago • 3 comments

hi： 支持一波，写的很棒，已经用上了，感谢。发现一个小问题，训练时 改变 train.sh gpus 参数 并不能指定 gpu。

jolt2017 avatar Jul 29 '21 08:07 jolt2017

哦，设置gpu的时候是用逗号,隔开的吗？gpus="3,4,5"，或者修改opt.gpus = "3,4,5"有效吗

zhangming8 avatar Jul 29 '21 14:07 zhangming8

hi： 支持一波，写的很棒，已经用上了，感谢。发现一个小问题，训练时 改变 train.sh gpus 参数 并不能指定 gpu。

这个bug现在已经修复，并且支持多gpu训练

zhangming8 avatar Aug 02 '21 16:08 zhangming8

hi： 支持一波，写的很棒，已经用上了，感谢。发现一个小问题，训练时 改变 train.sh gpus 参数 并不能指定 gpu。

这个bug现在已经修复，并且支持多gpu训练 好的 多谢

jolt2017 avatar Aug 03 '21 03:08 jolt2017