Skip to content

请问一下,有人遇到过这个问题吗,网上说是标签的越界,是在损失函数里面报的错。 #36

@ghost

Description

`RPN_REGR_Loss Exception: copy_if failed to synchronize: device-side assert triggered
Traceback (most recent call last):
File "./PyTorch/ctpn_/utils/loss/loss_utils.py", line 31, in forward
reg_keep = (cls == 1).nonzero()[:, 0]
RuntimeError: copy_if failed to synchronize: device-side assert triggered

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "./PyTorch/ctpn_/train.py", line 98, in
train()
File "./PyTorch/ctpn_/train.py", line 75, in train
loss_reg = criterion_reg(out_reg, reg_s)

loss = torch.tensor(0.0)

RuntimeError: CUDA error: device-side assert triggered`

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions