ValueError: ('Нераспознанные аргументы ключевого слова:', dict_keys (['input_shape'])) при загрузке модели - PullRequest
0 голосов
/ 23 апреля 2020

enter code here Я использую Google Colab, где использовать бесплатный GPU, требуется TenSenflow 2.0. Входной слой содержит 4 кортежа канала, который является несколькими 3D-изображениями. Я вызвал функцию:

    input_shape= input_shape=(4,96,96,96)
    model = build_model(input_shape=input_shape, output_channels=3)


   def build_model(input_shape=(4, 96,96,96), output_channels=3, weight_L2=0.1, weight_KL=0.1):
    ## Input Layer
   inp = tf.keras.Input(shape=[4,96,96,96],name='inp')
    ## The Initial Block
   x =  tf.keras.layers.Conv3D(
   filters=32,
   kernel_size=(3, 3, 3),
   strides=1,
   padding='same',
   data_format='channels_first',
   name='Input_x1')(inp)

   ## Dropout (0.2)
   x =  tf.keras.layers.Dropout(0.2)(x)

   ## Green Block x1 (output filters = 32)
   x1 = tf.green_block(x, tf.constant(32), name='x1')
   x =  tf.keras.layers.Conv3D(
   filters=32,
   kernel_size=(3, 3, 3),
   strides=2,
   padding='same',
   data_format='channels_first',
   name='Enc_DownSample_32')(x1)

Я получил следующая ошибка: ValueError Traceback (последний вызов был последним)

in ()

---> 89 модель = build_model (input_shape = input_shape, output_channels = 3) 90 91 model.fit (data , tags, batch_size = 1, epochs = 100)

1 фрейм

/ usr / local / lib / python3 .6 / dist-packages / tenorflow / python / keras / engine / input_layer.py в Input (shape, batch_size, name, dtype, sparse, тензорный, рваный, ** kwargs)

286   if kwargs:

-> 287 повысить ValueError ('Нераспознанные аргументы ключевого слова:', kwargs.keys ()) 288 289, если фигура равна None, а тензор равен None:

ValueError: ('Аргументы нераспознанного ключевого слова:', dict_keys (['input_shape']))

...