From 4bf8ddce83ada8c1d0281a8fe3bc330c42dfec1f Mon Sep 17 00:00:00 2001
From: Georgios Albanis <giorgos_al10@hotmail.com>
Date: Thu, 21 Mar 2024 11:33:10 +0200
Subject: [PATCH] feat: remove debug code

---
 moai/serve/streaming_optimizer.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/moai/serve/streaming_optimizer.py b/moai/serve/streaming_optimizer.py
index 24c11212..8487c9c8 100644
--- a/moai/serve/streaming_optimizer.py
+++ b/moai/serve/streaming_optimizer.py
@@ -301,10 +301,6 @@ def closure():
                 self.last_loss = None
                 self.optimizer.optimization_step = 0
             metrics = self.optimizer.validation(data)
-            # NOTE: DEBUG
-            if batch_idx >=3:
-                break
-            # NOTE: DEBUG
             # call postprocess
             log.info(f"Fitting Metrics: {metrics}") # TODO: why metrics still have gradients?
             # data["__moai__"]["batch_index"] = batch_idx