f4503bb3a3be014b452f54d8e2d187bb6419f627,tests/test_graph.py,,test_conv_wider,#,66

Before Change

def test_conv_wider():
    model = get_concat_skip_model()
    graph = Graph(model, True)
    graph.to_wider_model(5, 3)
    new_model = graph.produce_model()
    input_data = get_conv_data()

    output1 = model.predict_on_batch(input_data).flatten()
    output2 = new_model.predict_on_batch(input_data).flatten()

    assert np.sum(np.abs(output1 - output2)) < 4e-1

After Change

def test_conv_wider():
    graph = get_concat_skip_model()
    model = graph.produce_model()
    graph = deepcopy(graph)
    graph.to_wider_model(5, 3)
    new_model = graph.produce_model()
    input_data = torch.Tensor(get_conv_data())

    model.eval()
    new_model.eval()

    output1 = model(input_data)
    output2 = new_model(input_data)

Instances