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

Before Change

    with torch.no_grad():
        output_sru_250 = model(inputs)
        assert len(output_sru_235) == len(output_sru_250) == 2
        assert torch.allclose(output_sru_235[0], output_sru_250[0])
        assert torch.allclose(output_sru_235[1], output_sru_250[1])


if __name__ == "__main__":

After Change

        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