Skip to content
Snippets Groups Projects
Commit 61de6575 authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

fix unit tests

parent a592c144
Branches
Tags
1 merge request!24Deep mspad
...@@ -88,7 +88,7 @@ def test_architectures(): ...@@ -88,7 +88,7 @@ def test_architectures():
a = numpy.random.rand(1, 8, 224, 224).astype("float32") a = numpy.random.rand(1, 8, 224, 224).astype("float32")
t = torch.from_numpy(a) t = torch.from_numpy(a)
from ..architectures import DeepMSPAD from ..architectures import DeepMSPAD
net = FASNet(pretrained=False, num_channels=8) net = DeepMSPAD(pretrained=False, num_channels=8)
output = net.forward(t) output = net.forward(t)
assert output.shape == torch.Size([1, 1]) assert output.shape == torch.Size([1, 1])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment