clean up test_tp_fit

mosaicml · Sep 19, 2024 · 0255b6b · 0255b6b
1 parent 761001c
commit 0255b6b
Show file tree

Hide file tree

Showing 2 changed files with 134 additions and 234 deletions.
diff --git a/composer/models/tasks/classification.py b/composer/models/tasks/classification.py
@@ -133,6 +133,4 @@ def update_metric(self, batch: Any, outputs: Any, metric: Metric) -> None:
     def forward(self, batch: tuple[Tensor, Any]) -> Tensor:
         inputs, _ = batch
         outputs = self.module(inputs)
-        from icecream import ic
-        ic(inputs, outputs)
         return outputs