[test] added unit test for LightCNNs with 29 layers

bob/learn/pytorch/test/test.py

@@ -41,6 +41,24 @@ def test_architectures():
   output, emdedding = net.forward(t)
   assert output.shape == torch.Size([1, 79077])
   assert emdedding.shape == torch.Size([1, 256])
+  
+  # LightCNN29
+  a = numpy.random.rand(1, 1, 128, 128).astype("float32")
+  t = torch.from_numpy(a)
+  from ..architectures import LightCNN29
+  net = LightCNN29()
+  output, emdedding = net.forward(t)
+  assert output.shape == torch.Size([1, 79077])
+  assert emdedding.shape == torch.Size([1, 256])
+
+  # LightCNN29v2
+  a = numpy.random.rand(1, 1, 128, 128).astype("float32")
+  t = torch.from_numpy(a)
+  from ..architectures import LightCNN29v2
+  net = LightCNN29v2()
+  output, emdedding = net.forward(t)
+  assert output.shape == torch.Size([1, 79077])
+  assert emdedding.shape == torch.Size([1, 256])
 
   # DCGAN
   d = numpy.random.rand(1, 3, 64, 64).astype("float32")