From faa7e63639d00864a0e2ef197335265267dfcbaa Mon Sep 17 00:00:00 2001 From: Eduardo Leao Date: Sat, 20 Jul 2024 19:31:18 -0300 Subject: [PATCH] fix integration test --- tests/integration.test.ts | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/integration.test.ts b/tests/integration.test.ts index 7401e9a..4cdc6fa 100644 --- a/tests/integration.test.ts +++ b/tests/integration.test.ts @@ -29,11 +29,11 @@ function test_autograd(): number { let loss!: Tensor; // Instantiate Neural Network's Layers: - const w1 = randn([16, 32], true, true); + const w1 = randn([16, 32], true, 'cpu', true); const relu1 = new ReLU(); - const w2 = randn([32, 32], true, true); + const w2 = randn([32, 32], true, 'cpu', true); const relu2 = new ReLU(); - const w3 = randn([32, 50], true, true); + const w3 = randn([32, 50], true, 'cpu', true); // Training Loop: for (let i = 0; i < 128; i++) {