Skip to content

训练自己的数据时报错:an illegal memory access was encountered #241

Description

@fantasyfulan

我用create_dataset.py处理了自己的训练数据,然后放入CRNN中训练,出现以下错误:
/crnn_net/dataset.py:95: UserWarning: torch.range is deprecated in favor of torch.arange and will be removed in 0.5. Note that arange generates values in [start; end), not [start; end].
batch_index = random_start + torch.range(0, self.batch_size - 1)
THCudaCheck FAIL file=/pytorch/aten/src/THC/generic/THCTensorMath.cu line=26 error=77 : an illegal memory access was encountered
Traceback (most recent call last):
File "train.py", line 201, in
cost = trainBatch(crnn, criterion, optimizer)
File "train.py", line 187, in trainBatch
crnn.zero_grad()
File "/crnn_net/venv/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1114, in zero_grad
p.grad.zero_()
RuntimeError: cuda runtime error (77) : an illegal memory access was encountered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26
我不知道什么原因导致了这个错误,也不知道怎么处理,望告知,

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