DropPath
类keras_cv.layers.DropPath(rate=0.5, seed=None, **kwargs)
实现 DropPath 层。DropPath 在训练期间以 rate
的概率随机丢弃样本。请注意,此层丢弃批次中的单个样本,而不是整个批次。DropPath 随机丢弃批次中的一些单个样本,而 StochasticDepth 随机丢弃整个批次。
参考文献
参数
示例
DropPath
可在任何网络中使用,如下所示
# (...)
input = tf.ones((1, 3, 3, 1), dtype=tf.float32)
residual = keras.layers.Conv2D(1, 1)(input)
output = keras_cv.layers.DropPath()(input)
# (...)