feat: remove debug code

moverseai · Mar 21, 2024 · 4bf8ddc · 4bf8ddc
1 parent d25c114
commit 4bf8ddc
Showing 1 changed file with 0 additions and 4 deletions.
diff --git 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