@@ -4204,7 +4204,7 @@ class SequentialConvLayer {
4204
4204
}
4205
4205
4206
4206
4207
- } , 10 ) ;
4207
+ } , 50 ) ;
4208
4208
} ) ;
4209
4209
4210
4210
}
@@ -4354,7 +4354,7 @@ function convByOutputChannelAndInputSlicing(input, filter, biases, stride, pad,
4354
4354
res . layers [ i ] . dilationRate ,
4355
4355
3 ) ; // important for memory use
4356
4356
}
4357
-
4357
+
4358
4358
// Log memory usage
4359
4359
const memoryInfo = tf . memory ( ) ;
4360
4360
console . log ( `Iteration ${ i } :` ) ;
@@ -4429,7 +4429,7 @@ function convByOutputChannelAndInputSlicing(input, filter, biases, stride, pad,
4429
4429
// The second parameter is important for memory,
4430
4430
// the larger it is, the more memory it uses
4431
4431
// it was 8, but I set it to 3, got a different error
4432
- seqConvLayer = new SequentialConvLayer ( res , 4 , isChannelLast ) ;
4432
+ seqConvLayer = new SequentialConvLayer ( res , 3 , isChannelLast ) ;
4433
4433
4434
4434
// Apply the last output tensor to the seq. instance
4435
4435
let outputTensor = null ;
@@ -4853,7 +4853,7 @@ function convByOutputChannelAndInputSlicing(input, filter, biases, stride, pad,
4853
4853
//The second parameter is important for memory,
4854
4854
// the larger it is, the more memory it uses
4855
4855
// it was 8, but I set it to 3, got a different error
4856
- seqConvLayer = new SequentialConvLayer ( res , 4 , isChannelLast ) ;
4856
+ seqConvLayer = new SequentialConvLayer ( res , 3 , isChannelLast ) ;
4857
4857
4858
4858
4859
4859
// Apply the last output tensor to the seq. instance
0 commit comments