diff --git a/src/Model.js b/src/Model.js index 27c636d..d35442d 100644 --- a/src/Model.js +++ b/src/Model.js @@ -243,6 +243,9 @@ export default class Model { if ('activation' in wrappedLayerAttrs) { wrappedLayerAttrs.activation = camelCase(wrappedLayerAttrs.activation) } + if ('innerActivation' in wrappedLayerAttrs) { + wrappedLayerAttrs.innerActivation = camelCase(wrappedLayerAttrs.innerActivation) + } wrappedLayerAttrs.gpu = this.gpu layer = new layers[layerClass](Object.assign(attrs, { @@ -254,6 +257,9 @@ export default class Model { if ('activation' in attrs) { attrs.activation = camelCase(attrs.activation) } + if ('innerActivation' in attrs) { + attrs.innerActivation = camelCase(attrs.innerActivation) + } attrs.gpu = this.gpu layer = new layers[layerClass](attrs)