Closed ankitchandracs closed 2 years ago
code here is not using torch any more? What's your backend? Pytorch or MegEngine?
using it on google colab
Seems you are using Pytorch version code, please report your issue in another repo, Thx
did you solved it?]
File "/content/YOLOX/tools/train.py", line 115, in
assert num_gpu <= torch.cuda.device_count()
AssertionError