norse.torch.models.test