Function test_score

test/espnet2/asr/decoder/test_linear_decoder.py:24–28

(vocab_size, encoder_output_size, pooling)

@pytest.mark.parametrize("encoder_output_size", [4, 21])
@pytest.mark.parametrize("pooling", ["mean", "max", "CLS"])
def test_score(vocab_size, encoder_output_size, pooling):
    decoder = LinearDecoder(vocab_size, encoder_output_size, pooling)
    x = torch.randn(10, encoder_output_size)
    score, _ = decoder.score(ys=None, state=None, x=x)
    assert score.shape == (vocab_size,), score.shape