Source code for norse.torch.models.test.test_conv

import torch
from norse.torch.models import conv


[docs]def test_convnet4(): seq_length = 4 batch_size = 2 features = 1, 28, 28 model = conv.ConvNet4() x = torch.randn(seq_length, batch_size, *features) out = model(x) assert out.shape == torch.Size([seq_length, batch_size, 10])
[docs]def test_convnet(): seq_length = 4 batch_size = 2 features = 1, 28, 28 model = conv.ConvNet() x = torch.randn(seq_length, batch_size, *features) out = model(x) assert out.shape == torch.Size([seq_length, batch_size, 10])