AttributeError: 'list' object has no attribute 'split'

taynoel / YOLOv3-Mobilenetv2

Training and implementation program for light weight YOLOv3-MobileNet v2

GNU General Public License v3.0

25 stars 11 forks source link

AttributeError: 'list' object has no attribute 'split' #6

Open liupenglei opened 4 years ago

liupenglei commented 4 years ago

Traceback (most recent call last): File "trainMain.py", line 88, in a,label=torch.split(detections,[6,1],dim=1) File "/home/liupenglei/anaconda3/envs/pytorch1.1/lib/python3.6/site-packages/torch/functional.py", line 87, in split return tensor.split(split_size_or_sections, dim) AttributeError: 'list' object has no attribute 'split'