torch.cat不支持吗？unexpected keyword argument 'dim'

[wangshankun]

File "/home/shankun.shankunwan/CenterNet/src/lib/models/networks/DCNv2/dcn_v2.py", line 173, in forward offset = torch.cat((o1, o2), dim=1) File "/home/shankun.shankunwan/PytorchToCaffe/pytorch_to_caffe.py", line 534, in call out=self.obj(self.raw,*args,**kwargs) TypeError: _cat() got an unexpected keyword argument 'dim'

[JayLSU]

File "/home/shankun.shankunwan/CenterNet/src/lib/models/networks/DCNv2/dcn_v2.py", line 173, in forward offset = torch.cat((o1, o2), dim=1) File "/home/shankun.shankunwan/PytorchToCaffe/pytorch_to_caffe.py", line 534, in call out=self.obj(self.raw,*args,**kwargs) TypeError: _cat() got an unexpected keyword argument 'dim'

Change dim to dimension or just delete "dim=" will be fine

[OPPOA113]

@wangshankun hi 你那边CenterNet模型转caffe能跑通吗

[cwjhx]

@wangshankun hi 你那边CenterNet模型转caffe能跑通吗

我这边CenterNet去掉DCN模型转caffe可以转成功，但caffe转coreml时反卷积出错

[OPPOA113]

@cwjhx @wangshankun hi 你是直接去掉DCN还是讲DCN换成普通的conv？ 我这边是换成普通的conv之后训练的模型，在转成caffe 的时候是在base.level3.project.1的下一个batchnormal那出现错误，也没有提示什么错误，就中断了。。。 你那边是什么个情况呢？有没有修改什么的了。。。？方便加个qq：542854954 交流吗

[wangshankun]

@cwjhx @wangshankun hi 你是直接去掉DCN还是讲DCN换成普通的conv？ 我这边是换成普通的conv之后训练的模型，在转成caffe 的时候是在base.level3.project.1的下一个batchnormal那出现错误，也没有提示什么错误，就中断了。。。 你那边是什么个情况呢？有没有修改什么的了。。。？方便加个qq：542854954 交流吗

放弃caffe了，我转onnx，用tensorrt执行

[cwjhx]

@cwjhx @wangshankun hi 你是直接去掉DCN还是讲DCN换成普通的conv？ 我这边是换成普通的conv之后训练的模型，在转成caffe 的时候是在base.level3.project.1的下一个batchnormal那出现错误，也没有提示什么错误，就中断了。。。 你那边是什么个情况呢？有没有修改什么的了。。。？方便加个qq：542854954 交流吗

放弃caffe了，我转onnx，用tensorrt执行

我这边转ONNX还是出现RuntimeError: Only tuples, lists and Variables supported as JIT inputs, but got dict. 请问这个问题你那边是怎么解决的？ 谢谢

[wangshankun]

@cwjhx @wangshankun hi 你是直接去掉DCN还是讲DCN换成普通的conv？ 我这边是换成普通的conv之后训练的模型，在转成caffe 的时候是在base.level3.project.1的下一个batchnormal那出现错误，也没有提示什么错误，就中断了。。。 你那边是什么个情况呢？有没有修改什么的了。。。？方便加个qq：542854954 交流吗

放弃caffe了，我转onnx，用tensorrt执行

我这边转ONNX还是出现RuntimeError: Only tuples, lists and Variables supported as JIT inputs, but got dict. 请问这个问题你那边是怎么解决的？ 谢谢

好像在centernet的问题下回答了 https://github.com/xingyizhou/CenterNet/issues/77

[soham24]

@JayLSU should I change all dims in pytorch_to_caffe file to dimentions?

[happyday-lkj]

@JayLSU should I change all dims in pytorch_to_caffe file to dimentions?

just change the function _cat(raw, inputs, dimemsion) to _cat(raw, inputs, dim)

[sditeng]

改tools/pytorch_to_caffe.py文件中_cat

[zyc4me]

改tools/pytorch_to_caffe.py文件中_cat

改了之后会出现这个key error是为什么呢

conv = torch.cat((conv, feat_mid), dim=1)

File "./pytorch_to_caffe.py", line 594, in call out=self.obj(self.raw,*args,**kwargs) File "./pytorch_to_caffe.py", line 237, in _cat bottom_blobs.append(log.blobs(input)) File "./pytorch_to_caffe.py", line 88, in blobs print("{}:{} getting".format(var, self._blobs[var])) File "./pytorch_to_caffe.py", line 31, in getitem return self.data[key] KeyError: 140365443954224

[zyc4me]

改tools/pytorch_to_caffe.py文件中_cat

改了之后会出现这个key error是为什么呢

conv = torch.cat((conv, feat_mid), dim=1)

File "./pytorch_to_caffe.py", line 594, in call out=self.obj(self.raw,*args,kwargs) File "./pytorch_to_caffe.py", line 237, in _cat bottom_blobs.append(log.blobs(input)) File "./pytorch_to_caffe.py", line 88, in blobs print("{}:{} getting".format(var, self._blobs[var])) File "./pytorch_to_caffe.py", line 31, in getitem** return self.data[key] KeyError: 140365443954224

解决拉，是cat前一层的问题。。

[Liuzhenfa]

改tools/pytorch_to_caffe.py文件中_cat

改了之后会出现这个key error是为什么呢

conv = torch.cat((conv, feat_mid), dim=1)

File "./pytorch_to_caffe.py", line 594, in call out=self.obj(self.raw,*args,kwargs) File "./pytorch_to_caffe.py", line 237, in _cat bottom_blobs.append(log.blobs(input)) File "./pytorch_to_caffe.py", line 88, in blobs print("{}:{} getting".format(var, self._blobs[var])) File "./pytorch_to_caffe.py", line 31, in getitem** return self.data[key] KeyError: 140365443954224

解决拉，是cat前一层的问题。。

请问你是怎么解决的，我也碰到同样的问题

[kaaier]

您好，我转这层的时候修改了参数的问题，可还是转不过去是什么原因？非常期待您的帮助