From 3bead3cb645f9be83b0ba7b9033c4c00ee7ddea6 Mon Sep 17 00:00:00 2001
From: Eduardo Leao <eduardoleao@Eduardos-MacBook-Air.local>
Date: Tue, 13 Aug 2024 16:47:35 -0300
Subject: [PATCH] improve integration test #3

---
 tests/integration.test.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/integration.test.ts b/tests/integration.test.ts
index 4cdc6fa..4e12bf0 100644
--- 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: