search

xxradon / PytorchToCaffe

Pytorch model to caffe model, supported pytorch 0.3, 0.3.1, 0.4, 0.4.1 ,1.0 , 1.0.1 , 1.2 ,1.3 .notice that only pytorch 1.1 have some bugs
MIT License
783 stars 224 forks source link

split层不支持 #117

Open xingmurong opened 1 year ago

xingmurong commented 1 year ago

xfront, = torch.split(x1, [5, 1], dim=1)转caffe不支持类似于切片操作 如果用x_front = torch.index_select(x1, 1, torch.tensor([0, 1, 2, 3, 4]))，后一层则提醒TypeError: None has type NoneType, but expected one of: bytes, unicode 请教下怎么解决