我正在尝试将一些代码从tensorflow 1.x转换到tensorflow 2.x。到目前为止一切都很顺利,但我被困在了atrous卷积上。与其他层不同,似乎没有一对一的转换。有一个纯粹的keras实现here和一个tf.nn.atrous_conv2d实现here,但我也不确定是否可以在tf.keras.Model函数here中使用它们。b if bias else 0)
return out 我只想转换它,把它放
我有一个带有一些较大的135 x 135 x 1 x 3卷积滤波器的tensorflow模型。我发现tf.nn.conv2d对于如此大的过滤器变得不可用了--它试图使用超过60 it的内存,这时我需要关闭它。下面是重现我的错误的最小脚本:import numpy as np
frames, height, width, channels = 200, 321(注意:我还试着改变我<e