Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

caffemodel转tnnmodel不支持Power #61

Closed
mmx110 opened this issue Jun 15, 2020 · 3 comments
Closed

caffemodel转tnnmodel不支持Power #61

mmx110 opened this issue Jun 15, 2020 · 3 comments
Assignees
Labels
enhancement New feature or request

Comments

@mmx110
Copy link

mmx110 commented Jun 15, 2020

我看tnn支持的op里面有Power,我的caffe模型里有Power层,转换的时候报如下错:
Failed type not support: Power
应该怎么解决呢?

layer {
  name: "input"
  type: "Input"
  top: "data"
  input_param {
    shape {
      dim: 1
      dim: 1
      dim: 256
      dim: 256
    }
  }
}
layer {
  name: "trans"
  type: "Power"
  bottom: "data"
  top: "trans"
  power_param {
    power: 1.0
    scale: 0.0078125
    shift: -1.0
  }
}
@gttiankai gttiankai self-assigned this Jun 15, 2020
@gttiankai gttiankai added the enhancement New feature or request label Jun 15, 2020
@gttiankai
Copy link
Collaborator

有两种方案:
1. 由你来贡献下相关的转换代码。
2. 稍微等下,我后面支持下这个算子。

@gttiankai
Copy link
Collaborator

feature_support_caffe_power 分支已经可以支持power了,尝试下?
注意:docker 镜像还没有更新,如果要使用docker的方式,需要自己构建。

@gttiankai
Copy link
Collaborator

在master分支已经支持了

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants