如何将slim.repeat用于自定义图层块?

时间:2019-07-04 10:34:47

标签: tensorflow tensorflow-slim

我有一个自定义的卷积块,它是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。有关如何执行此操作的代码段将不胜感激。预先感谢。

0 个答案:

没有答案