Esempio n. 1
0
def test_RNNDecoder_invalid_type():
    with pytest.raises(ValueError):
        RNNDecoder(10, 12, rnn_type="foo")
Esempio n. 2
0
 def __init__(self, atype):
     super().__init__()
     self.decoder = RNNDecoder(50, 128, att_conf=dict(atype=atype))