6629a7801184d9ff646c266154a09e94d643c39d,test/test_regression_2.py,,run,#Any#,14

Before Change

    hidden_size = 304
    seq_len = 304

    inputs = torch.rand(seq_len, batch_size, input_size)
    model = sru.SRU(
        input_size,
        hidden_size,
        num_layers=4,

After Change

    with torch.no_grad():
        output_sru_250 = model(inputs)
    assert len(output_sru_235) == len(output_sru_250) == 2
    max_diff0 = (output_sru_235[0] - output_sru_250[0]).abs().max().item()
    max_diff1 = (output_sru_235[1] - output_sru_250[1]).abs().max().item()
    assert max_diff0 <= EPSILON
    assert max_diff1 <= EPSILON

Instances