move all layer params into attrs object

This commit is contained in:
Leon Chen
2016-09-18 23:20:59 -04:00
parent b1af2a7751
commit fd20f1b7b1
47 changed files with 285 additions and 162 deletions
+4 -4
View File
@@ -15,7 +15,7 @@ describe('convolutional layer: ZeroPadding3D', function () {
it(`[convolutional.ZeroPadding3D.0] padding 1,1,1 on 3x5x2x2 input, dimOrdering=tf`, function () {
const key = `convolutional.ZeroPadding3D.0`
console.log(`\n%c[${key}] padding 1,1,1 on 3x5x2x2 input, dimOrdering=tf`, styles.h3)
let testLayer = new layers.ZeroPadding3D([1, 1, 1], { dimOrdering: 'tf' })
let testLayer = new layers.ZeroPadding3D({ padding: [1, 1, 1], dimOrdering: 'tf' })
let t = new KerasJS.Tensor(TEST_DATA[key].input.data, TEST_DATA[key].input.shape)
console.log('%cin', styles.h4, stringifyCondensed(t.tensor))
const startTime = performance.now()
@@ -32,7 +32,7 @@ describe('convolutional layer: ZeroPadding3D', function () {
it(`[convolutional.ZeroPadding3D.1] padding 1,1,1 on 3x5x2x2 input, dimOrdering=th`, function () {
const key = `convolutional.ZeroPadding3D.1`
console.log(`\n%c[${key}] padding 1,1,1 on 3x5x2x2 input, dimOrdering=th`, styles.h3)
let testLayer = new layers.ZeroPadding3D([1, 1, 1], { dimOrdering: 'th' })
let testLayer = new layers.ZeroPadding3D({ padding: [1, 1, 1], dimOrdering: 'th' })
let t = new KerasJS.Tensor(TEST_DATA[key].input.data, TEST_DATA[key].input.shape)
console.log('%cin', styles.h4, stringifyCondensed(t.tensor))
const startTime = performance.now()
@@ -49,7 +49,7 @@ describe('convolutional layer: ZeroPadding3D', function () {
it(`[convolutional.ZeroPadding3D.2] padding 3,2,2 on 3x2x1x4 input, dimOrdering=tf`, function () {
const key = `convolutional.ZeroPadding3D.2`
console.log(`\n%c[${key}] padding 3,2,2 on 3x2x1x4 input, dimOrdering=tf`, styles.h3)
let testLayer = new layers.ZeroPadding3D([3, 2, 2], { dimOrdering: 'tf' })
let testLayer = new layers.ZeroPadding3D({ padding: [3, 2, 2], dimOrdering: 'tf' })
let t = new KerasJS.Tensor(TEST_DATA[key].input.data, TEST_DATA[key].input.shape)
console.log('%cin', styles.h4, stringifyCondensed(t.tensor))
const startTime = performance.now()
@@ -66,7 +66,7 @@ describe('convolutional layer: ZeroPadding3D', function () {
it(`[convolutional.ZeroPadding3D.3] padding 3,2,2 on 3x2x1x4 input, dimOrdering=th`, function () {
const key = `convolutional.ZeroPadding3D.3`
console.log(`\n%c[${key}] padding 3,2,2 on 3x2x1x4 input, dimOrdering=th`, styles.h3)
let testLayer = new layers.ZeroPadding3D([3, 2, 2], { dimOrdering: 'th' })
let testLayer = new layers.ZeroPadding3D({ padding: [3, 2, 2], dimOrdering: 'th' })
let t = new KerasJS.Tensor(TEST_DATA[key].input.data, TEST_DATA[key].input.shape)
console.log('%cin', styles.h4, stringifyCondensed(t.tensor))
const startTime = performance.now()