Keras 3 API 文档 / KerasCV / / 正则化层 / DropPath 层

DropPath 层

[源代码]

DropPath

keras_cv.layers.DropPath(rate=0.5, seed=None, **kwargs)

实现 DropPath 层。DropPath 在训练期间以 rate 的概率随机丢弃样本。请注意,此层丢弃批次中的单个样本,而不是整个批次。DropPath 随机丢弃批次中的一些单个样本,而 StochasticDepth 随机丢弃整个批次。

参考文献

参数

  • rate: 浮点数,残差分支被丢弃的概率。
  • seed: (可选)整数。用于创建随机种子。

示例

DropPath 可在任何网络中使用,如下所示

# (...)
input = tf.ones((1, 3, 3, 1), dtype=tf.float32)
residual = keras.layers.Conv2D(1, 1)(input)
output = keras_cv.layers.DropPath()(input)
# (...)