improve integration test #3

eduardoleao052 · Aug 13, 2024 · 3bead3c · 3bead3c
1 parent fc16e2e
commit 3bead3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration.test.ts b/tests/integration.test.ts
@@ -196,7 +196,7 @@ function test_transformer(): number {
   let loss!: Tensor;
 
   // Training Loop:
-  for (let i = 0; i < 50; i++) {
+  for (let i = 0; i < 75; i++) {
     const z = model.forward(x);
 
     // Get loss: