wassname/keras-js
1
0
Fork 0
mirror of https://github.com/wassname/keras-js.git synced 2026-09-10 12:15:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

camelCase innerActivation if present

Browse Source
This commit is contained in:
Leon Chen
2016-10-13 00:32:11 -04:00
parent ccf94b1d3f
commit af1dcfa925
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Model.js
+6
View File
@@ -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)