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

如何修改caffemodel？ #89

Open 205418367 opened 4 years ago

205418367 commented 4 years ago

我想在caffemodel中添加新的层，比如softmax等，有什么方法去实现呢？我看到Caffe目录下有一些代码，我尝试了一下，添加的结果不准确。希望帮忙解决一下这个问题。

ShenDC commented 4 years ago

如果是无参数层，直接修改你的caffe模型的prototxt文件会更方便一些。注意该新增层的输出输出尺寸。

205418367 commented 4 years ago

已经解决