From c8eb01ca92214fc2f11eb49ab63cbc18d7dd35ff Mon Sep 17 00:00:00 2001
From: lucasdelimanogueira <lucasln96@hotmail.com>
Date: Fri, 31 May 2024 17:37:09 -0300
Subject: [PATCH] test cloud multigpu

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 22a04c5..ff71c50 100644
--- a/train.py
+++ b/train.py
@@ -111,9 +111,9 @@ def main2():
 
 
             optimizer.step()
+            break
 
-        print(f'Epoch [{epoch + 1}/{epochs}], Loss: {loss[0]:.4f}')
-        loss_list.append(loss[0])
+        break
 
 if __name__ == "__main__":
     main()