acheng777
acheng777
root@autodl-container-809011af9e-2908fc6e:~/edgeyolo# python detect.py --weights ./weights/edgeyolo_coco.pth --source test.avi --fp16 2023-09-01 18:00:05.169 | INFO | edgeyolo.models:__init__:50 - loading models from weight /root/edgeyolo/weights/edgeyolo_coco.pth Params: 41.23M, Gflops: 126.71 Reparameterizing models... After re-parameterization: Params: 40.51M,...
CUDA问题
报错信息如下: root@autodl-container-7154118f52-44680f8e:~/edgeyolo-main# python train.py --cfg ./params/train/train_coco.yaml Traceback (most recent call last): File "train.py", line 16, in train("DEFAULT" if args.default else args.cfg) File "/root/edgeyolo-main/edgeyolo/train/launch_train.py", line 101, in launch mp.start_processes( File "/root/miniconda3/lib/python3.8/site-packages/torch/multiprocessing/spawn.py",...