From af1dcfa9253293a5e47493527816d7d033e7493d Mon Sep 17 00:00:00 2001 From: Leon Chen Date: Thu, 13 Oct 2016 00:32:11 -0400 Subject: [PATCH] camelCase innerActivation if present --- src/Model.js | 6 ++++++ 1 file changed, 6 insertions(+) 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)