71a791cb71ef7080ea150152633a90e7821070af,tests/keras/applications/applications_test.py,,test_inceptionresnetv2_variable_input_channels,#,211

Before Change

    global_image_data_format = K.image_data_format()

    K.set_image_data_format("channels_first")
    input_shape = (1, None, None)
    model = applications.InceptionResNetV2(weights=None, include_top=False, input_shape=input_shape)
    assert model.output_shape == (None, 1536, None, None)
    input_shape = (4, None, None)
    model = applications.InceptionResNetV2(weights=None, include_top=False, input_shape=input_shape)
    assert model.output_shape == (None, 1536, None, None)

    K.set_image_data_format("channels_last")

After Change

        model = applications.InceptionResNetV2(weights=None, include_top=False, input_shape=input_shape)
        queue.put(model.output_shape)

    queue = Queue()
    p = Process(target=target, args=(queue, (None, None, 1)))
    p.start()
    p.join()
    assert not queue.empty(), "Model creation failed."
    model_output_shape = queue.get_nowait()
    assert model_output_shape == (None, None, None, 1536)

    p = Process(target=target, args=(queue, (None, None, 4)))
    p.start()

Instances