76b07a3212541102e66f4aa91335c68fac8ef00b,tests/test_net_transformer.py,,test_default_transform,#,48

Before Change

def test_default_transform():
    graphs = default_transform(DefaultClassifierGenerator(10, (28, 28, 1)).generate())
    // print()
    // for index, layer in enumerate(graphs[0].layer_list):
    //     print(index, layer)
    graphs[0].produce_model()
    assert len(graphs) == 1
    assert len(graphs[0].layer_list) == 42

After Change

def test_default_transform():
    graphs = default_transform(DefaultClassifierGenerator(10, (32, 32, 3)).generate())
    model = graphs[0].produce_model()
    model(torch.Tensor(get_conv_data()))
    assert len(graphs) == 1
    assert len(graphs[0].layer_list) == 42

Instances