1 报错描述
1.1 系统环境
Environment(Ascend/GPU/CPU): GPU-GTX3090(24G)
Software Environment:
– MindSpore version (source or binary): 1.7.0
– Python version (e.g., Python 3.7.5): 3.8.13
– OS platform and distribution (e.g., Linux Ubuntu 16.04): Ubuntu 16.04
– CUDA version : 11.0
1.2 基本信息
1.2.1脚本
此代码是ConvLSTM从PyTorch迁移到MindSpore的一部分,下面为报错部分
split = ops.Split(1, 2)
output = split(x)
1.2.2报错
部分个人信息做遮挡处理
Traceback (most recent call last):
File "main.py", line 195, in <module>
train()
File "main.py", line 142, in train
loss = train_network(data, label)
File "/home/xxxlab/anaconda2/envs/mindspore/lib/python3.8/site-packages/mindspore/nn/cell.py", line 612, in __call__
raise err
File "/home/xxxlab/anaconda2/envs/mindspore/lib/python3.8/site-packages/mindspore/nn/cell.py", line 609, in __call__
output = self._run_construct(cast_inputs, kwargs)
File "/home/xxxlab/anaconda2/envs/mindspore/lib/python3.8/site-packages/mindspore/nn/cell.py", line 429, in _run_const
ruct
output = self.construct(*cast_inputs, **kwargs)
File "/home/xxxlab/anaconda2/envs/mindspore/lib/python3.8/site-packages/mindspore/nn/wrap/cell_wrapper.py", line 373,
in construct
loss = self.network(*inputs)
File "