|
|
@@ -293,7 +293,7 @@ ncnn.Node = class {
|
|
|
break;
|
|
|
}
|
|
|
case 'InnerProduct': {
|
|
|
- const activation_names = [ '', 'ReLU', 'Leaky ReLU', 'Clip', 'Sigmoid', 'Mish' ];
|
|
|
+ const activation_names = [ '', 'ReLU', 'Leaky ReLU', 'Clip', 'Sigmoid', 'Mish', 'HardSwish' ];
|
|
|
const activation_type = parseInt(attributes.get('9') || 0, 10);
|
|
|
if (activation_type > 0 && activation_type < activation_names.length) {
|
|
|
const layer = {
|
|
|
@@ -330,7 +330,7 @@ ncnn.Node = class {
|
|
|
case 'ConvolutionDepthWise':
|
|
|
case 'Deconvolution':
|
|
|
case 'DeconvolutionDepthWise': {
|
|
|
- const activation_names = [ '', 'ReLU', 'LeakyReLU', 'Clip', 'Sigmoid', 'Mish' ];
|
|
|
+ const activation_names = [ '', 'ReLU', 'LeakyReLU', 'Clip', 'Sigmoid', 'Mish', 'HardSwish' ];
|
|
|
const activation_type = parseInt(attributes.get('9') || 0, 10);
|
|
|
if (activation_type > 0 && activation_type < activation_names.length) {
|
|
|
const layer = {
|