训练期间出错：布局失败：INVALID_ARGUMENT：值 0 的大小与排列 4 的大小不匹配

Question

我正在使用TensorFlow训练分割模型，在训练过程中遇到错误。大约 6 秒后，训练停止并显示以下错误消息：

Epoch 1/100
2023-07-17 08:14:20.618828: E tensorflow/core/grappler/optimizers/meta_optimizer.cc:954] layout failed: INVALID_ARGUMENT: Size of values 0 does not match size of permutation 4 @ fanin shape inmodel_3/dropout_15/dropout/SelectV2-2-TransposeNHWCToNCHW-LayoutOptimizer
8278/8278 [==============================] - 6s 198us/step - loss: 2.1831 - accuracy: 0.8421 - val_loss: 2.2880 - val_accuracy: 0.8349

我正在使用自定义数据生成器 ( DataGen) 来加载和预处理输入图像和蒙版。该错误似乎与模型的布局有关，特别是图层dropout。我不确定为什么值的大小与排列大小不匹配。我认为这可能与数据生成器有关。

我在下面包含了相关的代码片段：

Epoch 1/100
2023-07-17 08:14:20.618828: E tensorflow/core/grappler/optimizers/meta_optimizer.cc:954] layout failed: INVALID_ARGUMENT: Size of values 0 does not match size of permutation 4 @ fanin shape inmodel_3/dropout_15/dropout/SelectV2-2-TransposeNHWCToNCHW-LayoutOptimizer
8278/8278 [==============================] - 6s 198us/step - loss: 2.1831 - accuracy: 0.8421 - val_loss: 2.2880 - val_accuracy: 0.8349

任何有关如何解决此问题的见解或建议将不胜感激。

我正在使用 coco2014 数据集。tf 版本“2.12.0”

Answer 1

我的训练器的第一个时期有一个非常相似的错误消息：

tensorflow/core/grappler/optimizers/meta_optimizer.cc:961] layout failed: INVALID_ARGUMENT: Size of values 0 does not match size of permutation 4 @ fanin shape insequential_2/efficientnetv2-b0/block2b_drop/dropout/SelectV2-2-TransposeNHWCToNCHW-LayoutOptimizer

我了解到，就我而言，我可以忽略这一点。适合例程继续工作并且工作正常......