小编51s*_*sep的帖子

我是初学者。我尝试了 Tensorflow 的图像分类，并得到以下错误。我在网上发现了类似的问题，但我无法理解。错误是什么意思？我该怎么办？请给我一些建议。我使用 100 个文件（png/15pix、15pix）作为示例图像。Tensorflow ver.2.0.0 / python ver.3.8.1 / Jupyter notebook。

    num_epochs = 30
    steps_per_epoch = round(num_train)//BATCH_SIZE
    val_steps = 20
    history = model.fit(train_data.repeat(),
                epochs=num_epochs,
                steps_per_epoch = steps_per_epoch,
                validation_data=val_data.repeat(), 
                validation_steps=val_steps)

InvalidArgumentError：输入深度必须被过滤器深度整除：4 vs 3 [[节点序列_2/mobilenetv2_1.00_96/Conv1/Conv2D（定义在C:\Users\XXXXX\Anaconda3\lib\site-packages\tensorflow_core\python\framework \ops.py:1751) ]] [Op:__inference_distributed_function_42611] 函数调用堆栈：distributed_function