Replies: 1 comment 10 replies
-
Hi, @chintha-3! 👋🏻 Thanks for moving that question from YouTube to GitHub. Is that error in our official notebook, or is that your custom code? |
Beta Was this translation helpful? Give feedback.
10 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
Traceback (most recent call last):
File "train.py", line 616, in
train(hyp, opt, device, tb_writer)
File "train.py", line 363, in train
loss, loss_items = compute_loss_ota(pred, targets.to(device), imgs) # loss scaled by batch_size
File "/content/drive/MyDrive/Final year project/yolov7/utils/loss.py", line 585, in call
bs, as_, gjs, gis, targets, anchors = self.build_targets(p, targets, imgs)
File "/content/drive/MyDrive/Final year project/yolov7/utils/loss.py", line 753, in build_targets
_, cost_argmin = torch.min(cost[:, anchor_matching_gt > 1], dim=0)
RuntimeError: CUDA error: device-side assert triggered
Beta Was this translation helpful? Give feedback.
All reactions