-
Notifications
You must be signed in to change notification settings - Fork 124
Open
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`
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels