3a431ea52d090fb3ef8a1e0e5d7f796d9a42e097,tests/keras/applications/applications_test.py,,test_inceptionresnetv2_notop,#,198

Before Change

        queue.put(model.output_shape)

    global_image_data_format = K.image_data_format()
    queue = Queue()

    K.set_image_data_format("channels_first")
    p = Process(target=target, args=(queue,))
    p.start()
    p.join()
    K.set_image_data_format(global_image_data_format)
    assert not queue.empty(), "Model creation failed."
    model_output_shape = queue.get_nowait()
    assert model_output_shape == (None, 1536, None, None)

    K.set_image_data_format("channels_last")
    p = Process(target=target, args=(queue,))
    p.start()
    p.join()
    K.set_image_data_format(global_image_data_format)
    assert not queue.empty(), "Model creation failed."
    model_output_shape = queue.get_nowait()
    assert model_output_shape == (None, None, None, 1536)


@keras_test

After Change

def test_inceptionresnetv2_notop():
    def model_fn():
        return applications.InceptionResNetV2(weights=None, include_top=False)
    output_shape = clean_run(model_fn)
    if K.image_data_format() == "channels_first":
        assert output_shape == (None, 1536, None, None)
    else:
        assert output_shape == (None, None, None, 1536)

Instances