convert to onnx is error

[dengfenglai321]

hi
could you tell me how to convert to onnx.
I convert to onnx and it occur error as below:
File "/data2/multimedia_gpu03/anaconda3/lib/python3.6/site-packages/torch/functional.py", line 78, in split
return tensor.split(split_size_or_sections, dim)
File "/data2/multimedia_gpu03/anaconda3/lib/python3.6/site-packages/torch/tensor.py", line 308, in split
return super(Tensor, self).split_with_sizes(split_size, dim)
TypeError: split_with_sizes(): argument 'split_sizes' (position 1) must be tuple of ints, not Tensor

[NetEase-GameAI]

It's more like a PyTorch issue.
Try n, c, h, w = tensor.size(), then use these n, c, h, w ints during split.