使用具有可变批处理大小的tf.contrib.layers.group_norm的问题

Question

我正在尝试将群组规范用作CNN的一部分。但是我收到以下错误，我认为这与以下事实有关：我输入图像的占位符的第一维（即批处理大小）设置为“无”。组规范的实现涉及一个重整操作，不能很好地解决此问题。您能建议一个解决方法吗？

TypeError：无法将类型的对象转换为Tensor。内容：[无，16、64、64、12、2]。考虑将元素强制转换为受支持的类型。

感谢您的时间和精力。

Answer 1

我相信这是Tensorflow上的一个已知问题，您可以在此处阅读有关此内容的更多信息：https://github.com/tensorflow/tensorflow/issues/24409

简而言之，此错误是由于normalization.py#L303引起的，可以通过在normalization.py中的第252行之后添加以下两行来解决：

original_shape = [-1 if s.value is None else s for s in original_shape]
if len([s for s in original_shape if s == -1]) > 1:
    raise ValueError('Only one axis dimension can be undefined in the input tensor')

和第302行之后的以下行：

inputs_shape = [-1 if s is None else s for s in inputs_shape]

来源：This在tensorflow github页面上。