Replies: 1 comment
-
Oops, i tried to change all cuda() to cpu() and it worked |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I tried to change file train_det.py (36 line) to
But in file yolov7.py there's
mask, obj_mask = mask.cuda(), obj_mask.cuda()
Which kills option to try it with CPU
I want to test my M1
Beta Was this translation helpful? Give feedback.
All reactions