yolov3-pytorch训练报错

OngoingDC 2020-11-11 04:07:31
/pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:60: lambda [](int)->auto::operator()(int)->auto: block: [0,0,0], thread: [41,0,0] Assertion `index >= -sizes[i] && index < sizes[i] && "index out of bounds"` failed.
/pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:60: lambda [](int)->auto::operator()(int)->auto: block: [0,0,0], thread: [45,0,0] Assertion `index >= -sizes[i] && index < sizes[i] && "index out of bounds"` failed.
/pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:60: lambda [](int)->auto::operator()(int)->auto: block: [0,0,0], thread: [47,0,0] Assertion `index >= -sizes[i] && index < sizes[i] && "index out of bounds"` failed.
Traceback (most recent call last):
File "train.py", line 105, in <module>
loss, outputs = model(imgs, targets)
File "/home/ducong/anaconda3/envs/yolov3-pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__
result = self.forward(*input, **kwargs)
File "/home/ducong/ducong/YOLOv3/PyTorch-YOLOv3/models.py", line 260, in forward
x, layer_loss = module[0](x, targets, img_dim)
File "/home/ducong/anaconda3/envs/yolov3-pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__
result = self.forward(*input, **kwargs)
File "/home/ducong/ducong/YOLOv3/PyTorch-YOLOv3/models.py", line 188, in forward
ignore_thres=self.ignore_thres,
File "/home/ducong/ducong/YOLOv3/PyTorch-YOLOv3/utils/utils.py", line 306, in build_targets
noobj_mask[b[i], anchor_ious > ignore_thres, gj[i], gi[i]] = 0
RuntimeError: CUDA error: device-side assert triggered


这是咋了啊,不太懂,有知道的吗
...全文
3551 2 打赏 收藏 转发到动态 举报
AI 作业
写回复
用AI写文章
2 条回复
切换为时间正序
请发表友善的回复…
发表回复
OngoingDC 2021-04-04
  • 打赏
  • 举报
回复
引用 1 楼 weixin_43390800 的回复:
博主你好,我也遇到这个问题了,请问你解决了吗?
忘记了怎么操作的
weixin_43390800 2021-04-04
  • 打赏
  • 举报
回复
博主你好,我也遇到这个问题了,请问你解决了吗?

357

社区成员

发帖
与我相关
我的任务
社区描述
CUDA高性能计算讨论
社区管理员
  • CUDA高性能计算讨论社区
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧