fix ci

Signed-off-by: KumoLiu <[email protected]>
Project-MONAI · Sep 1, 2023 · 954332c · 954332c
1 parent b1fb99f
commit 954332c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_dice_ce_loss.py b/tests/test_dice_ce_loss.py
@@ -97,7 +97,7 @@ def test_ill_reduction(self):
 
     def test_script(self):
         loss = DiceCELoss()
-        test_input = torch.ones(2, 1, 8, 8)
+        test_input = torch.ones(2, 2, 8, 8)
         test_script_save(loss, test_input, test_input)