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

update convolution2d layer

Browse Source
This commit is contained in:
Leon Chen
2016-08-26 13:23:31 -04:00
parent efb2280e2c
commit 5fa5fb136b
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/layers/convolutional/Convolution2D.js
+4 -4
View File
@@ -31,7 +31,7 @@ export default class Convolution2D extends Layer {
this.activation = activations[activation]
if (['valid', 'same'].indexOf(borderMode) > -1) {
if (borderMode === 'valid' || borderMode === 'same') {
this.borderMode = borderMode
} else {
throw new Error(`${this.name} [Convolution2D layer] Invalid borderMode.`)
@@ -39,10 +39,10 @@ export default class Convolution2D extends Layer {
this.subsample = subsample
if (['tf', 'th'].indexOf(dimOrdering) > -1) {
this.dimOrdering = dimOrdering
if (dimOrdering !== 'tf') {
throw new Error(`${this.name} [Convolution2D layer] Only tf dim ordering supported currently.`)
} else {
throw new Error(`${this.name} [Convolution2D layer] Invalid dimOrdering.`)
this.dimOrdering = dimOrdering
}
this.bias = bias