HomePod上训练“TEXT CLASSIFICATION WITH THE TORCHTEXT LIBRARY”例子报错

代码为：https://pytorch.org/tutorials/beginner/text_sentiment_ngrams_tutorial.html
环境为：MLab HomePod 2.0 pro

报错信息为：

# 在cpu上时
root@homepod:/.gemfield_install# python train.py 
/opt/conda/lib/python3.8/site-packages/torch/nn/init.py:388: UserWarning: Initializing zero-element tensors is a no-op
warnings.warn("Initializing zero-element tensors is a no-op")
test.csv: 1.86MB [00:00, 59.1MB/s]                                                                                                                                                                                                                           
Traceback (most recent call last):
File "train.py", line 118, in <module>
train(train_dataloader)
File "train.py", line 68, in train
loss = criterion(predited_label, label)
File "/opt/conda/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "/opt/conda/lib/python3.8/site-packages/torch/nn/modules/loss.py", line 1120, in forward
return F.cross_entropy(input, target, weight=self.weight,
File "/opt/conda/lib/python3.8/site-packages/torch/nn/functional.py", line 2824, in cross_entropy
return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
IndexError: Target 2 is out of bounds.

当在cuda设备上时：


Traceback (most recent call last):
File "train.py", line 120, in <module>
train(train_dataloader)
File "train.py", line 71, in train
loss.backward()
File "/opt/conda/lib/python3.8/site-packages/torch/_tensor.py", line 255, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph, inputs=inputs)
File "/opt/conda/lib/python3.8/site-packages/torch/autograd/__init__.py", line 147, in backward
Variable._execution_engine.run_backward(
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
/opt/conda/conda-bld/pytorch_1623448278899/work/aten/src/THCUNN/ClassNLLCriterion.cu:108: cunn_ClassNLLCriterion_updateOutput_kernel: block: [0,0,0], thread: [0,0,0] Assertion `t >= 0 && t < n_classes` failed.
/opt/conda/conda-bld/pytorch_1623448278899/work/aten/src/THCUNN/ClassNLLCriterion.cu:108: cunn_ClassNLLCriterion_updateOutput_kernel: block: [0,0,0], thread: [1,0,0] Assertion `t >= 0 && t < n_classes` failed.
.......
/opt/conda/conda-bld/pytorch_1623448278899/work/aten/src/THCUNN/ClassNLLCriterion.cu:108: cunn_ClassNLLCriterion_updateOutput_kernel: block: [0,0,0], thread: [31,0,0] Assertion `t >= 0 && t < n_classes` failed.

DeepVAC / MLab

HomePod上训练“TEXT CLASSIFICATION WITH THE TORCHTEXT LIBRARY”例子报错 #53