UnimplementedError:融合的conv实现暂时不支持分组卷积

问题描述

我正在尝试使用TU-Berlin构建CNN模型以识别人体草图数据集.我下载了png zip文件，将数据导入了Google Colab，然后将数据拆分为训练测试文件夹.这是模型:

I am trying to build a CNN model to recognise human sketch using the TU-Berlin dataset. I downloaded the png zip file, imported the data to Google Colab and then split the data into train-test folders. Here is the model:

model = tf.keras.models.Sequential([
    tf.keras.layers.Conv2D(filters = 64, kernel_size = (5,5),padding = 'Same', 
                 activation ='relu', input_shape = target_dims),
    tf.keras.layers.Conv2D(filters = 64, kernel_size = (5,5),padding = 'Same', 
                 activation ='relu'),
    tf.keras.layers.MaxPool2D(pool_size=(2,2)),
    tf.keras.layers.Dropout(0.25),

    tf.keras.layers.Conv2D(filters = 128, kernel_size = (3,3),padding = 'Same', 
                 activation ='relu'),
    tf.keras.layers.Conv2D(filters = 128, kernel_size = (3,3),padding = 'Same', 
                 activation ='relu'),
    tf.keras.layers.MaxPool2D(pool_size=(2,2), strides=(2,2)),
    tf.keras.layers.Dropout(0.25),

    tf.keras.layers.Conv2D(256, kernel_size=4, strides=1, activation='relu', padding='same'),
    tf.keras.layers.Conv2D(256, kernel_size=4, strides=2, activation='relu', padding='same'),
    tf.keras.layers.Dropout(0.25),

    tf.keras.layers.Flatten(),
    tf.keras.layers.Dense(512, activation = "relu"),
    tf.keras.layers.Dropout(0.5),
    tf.keras.layers.Dense(n_classes, activation= "softmax")
])

model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=["accuracy"])

model.fit_generator(train_generator, epochs=10, validation_data=val_generator)

我收到以下错误:

UnimplementedError:  Fused conv implementation does not support grouped convolutions for now.
     [[node sequential/conv2d/Relu (defined at <ipython-input-9-36d4624b896d>:1) ]] [Op:__inference_train_function_1358]

Function call stack:
train_function

对于能解决此问题的任何帮助，我将不胜感激.谢谢你.

I would be grateful to any kind of help that will solve this issue. Thank you.

(PS-我正在运行Tensorflow 2.2.0，没有GPU)

(PS - I am running Tensorflow 2.2.0 and no GPU)

推荐答案

我遇到了类似的错误，问题出在我的图像的通道数和我在模型中指定的通道数.因此，请检查图像的尺寸数并检查输入形状中指定的值，以确保它们相同

I had a similar error, the problem was with the number of channels for my image and the number of channels I specified in the model. So check the number of dimension of your image and check the value specified in the input shape ensure they are the same

这篇关于UnimplementedError:融合的conv实现暂时不支持分组卷积的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！