完整报错:
Traceback (most recent call last):
File “train.py”, line 73, in
feature = model(seq)
File “/home/liang/miniconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py”, line 541, in
call
result = self.forward(*input, *
kwargs)
File “/home/liang/miniconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py”, line 148, in forward
inputs, kwargs = self.scatter(inputs, kwargs, self.device_ids)
File “/home/liang/miniconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py”, line 159, in scatter
return scatter_kwargs(inputs, kwargs, device_ids, dim=self.dim)
File “/home/liang/miniconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/parallel/scatter_gather.py”, line 36, in scatter_kwargs
inputs = scatter(inputs, target_gpus, dim) if inputs else []
File “/home/liang/miniconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/parallel/scatter_gather.py”, line 28, in scatter
res = scatter_map(inputs)
File “/home/liang/miniconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/parallel/scatter_gather.py”, line 15, in scatter_map
return list(zip(
map(scatter_map, obj)))
File “/home/liang/miniconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/parallel/scatter_gather.py”, line 13, in scatter_map
return Scatter.apply(target