nvdla / hw

RTL, Cmodel, and testbench for NVDLA
Other
1.67k stars 561 forks source link

Problem with PReLU #359

Closed timzhang32 closed 2 years ago

timzhang32 commented 2 years ago

Hi everyone,

seems like NVDLA can't execute PReLU correctly by default.

I have the following description in prototxt: layer { name: "relu1" type: "ReLU" bottom: "scale1" top: "relu1" relu_param { negative_slope: 0.1 } }

Although it can be parsed by the compiler, on NVDLA it's being executed as a standard ReLU. Does anyone know how to change the KMD, in order to properly execute this layer as PReLU?

Thanks.

Best Tim