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
-
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