unit test prints

Deci-AI · Jan 5, 2023 · 56d038e · 56d038e
1 parent b6d47b8
commit 56d038e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/unit_tests/repvgg_unit_test.py b/tests/unit_tests/repvgg_unit_test.py
@@ -72,6 +72,8 @@ def test_deployment_architecture(self):
 
             deployment_mode_output = model(test_input)
             # difference is of very low magnitude
+            print("training_mode_output: " + str(training_mode_output))
+            print("deployment_mode_output: " + str(deployment_mode_output))
             self.assertFalse(False in torch.isclose(training_mode_output, deployment_mode_output, atol=1e-6))
 
     def test_backbone_mode(self):