🐛 Remove test case bug

rishikksh20 · Aug 31, 2020 · 67f5629 · 67f5629
1 parent 75b32eb
commit 67f5629
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_res_unet.py b/tests/test_res_unet.py
@@ -3,7 +3,7 @@
 
 def test_resunet():
     img = torch.ones(1, 3, 224, 224)
-    resunet = ResUnet(3, 64)
+    resunet = ResUnet(3)
     assert resunet(img).shape == torch.Size([1, 1, 224, 224]) 
 
 
@@ -16,4 +16,5 @@ def test_residual_conv():
 def test_upsample():
     x = torch.ones(1, 512, 28, 28)
     upsample = Upsample(512, 512, 2, 2)
-    assert upsample(x).shape == torch.Size([1, 512, 56, 56])
+    assert upsample(x).shape == torch.Size([1, 512, 56, 56])
+