我有一个自定义的卷积块,它是conv2d和可分离的卷积层的集合,有点像这样:
def custom_block():
output = conv2d(inputs, scope="conv")
output = nn.separable(output, scope=conv2)
我需要重复这一卷积步骤。为此,我正在使用tf.contrib.slim.repeat(custom_block, scope=custom)
但是我面对这样的错误:
ValueError: Variable custom/conv/beta already exists, disallowed. Did you mean to set reuse=True or reuse=tf.AUTO_REUSE in VarScope? Originally defined at
我搜索了许多contrib.repeat示例,但其中大多数都使用了已经在tensorflow lib中定义的conv块,没有一个自定义块。 我不确定是否需要将自定义conv的两个conv包装到变量范围中,并使用复用= tf.AUTO_REUSE。有关如何执行此操作的代码段将不胜感激。预先感谢。