DropPath 層

[來源]

DropPath 類別

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

實作 DropPath 層。 DropPath 在訓練期間以 rate 的機率隨機丟棄樣本。 請注意，此層會丟棄批次內的個別樣本，而不是整個批次。 DropPath 會從批次中隨機丟棄一些個別樣本，而 StochasticDepth 會隨機丟棄整個批次。

參考文獻

• FractalNet。 - rwightman/pytorch-image-models

參數

• 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)
# (...)