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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

split层不支持

xingmurong opened this issue · comments

x_front, _ = 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
请教下怎么解决