Skip to content

Commit 87ebc94

Browse files
committed
fix memory
1 parent 8e45ec7 commit 87ebc94

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

js/brainchop/mainMeshNetFunctions.js

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4204,7 +4204,7 @@ class SequentialConvLayer {
42044204
}
42054205

42064206

4207-
}, 10);
4207+
}, 50);
42084208
});
42094209

42104210
}
@@ -4354,7 +4354,7 @@ function convByOutputChannelAndInputSlicing(input, filter, biases, stride, pad,
43544354
res.layers[i].dilationRate,
43554355
3); // important for memory use
43564356
}
4357-
4357+
43584358
// Log memory usage
43594359
const memoryInfo = tf.memory();
43604360
console.log(`Iteration ${i}:`);
@@ -4429,7 +4429,7 @@ function convByOutputChannelAndInputSlicing(input, filter, biases, stride, pad,
44294429
// The second parameter is important for memory,
44304430
// the larger it is, the more memory it uses
44314431
// 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);
44334433

44344434
// Apply the last output tensor to the seq. instance
44354435
let outputTensor = null;
@@ -4853,7 +4853,7 @@ function convByOutputChannelAndInputSlicing(input, filter, biases, stride, pad,
48534853
//The second parameter is important for memory,
48544854
// the larger it is, the more memory it uses
48554855
// 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);
48574857

48584858

48594859
// Apply the last output tensor to the seq. instance

0 commit comments

Comments
 (0)