mmlearn.modules.layers.transformer_block.DropPath¶

class DropPath(drop_prob=0.0)[source]¶

Bases: Module

Drop paths (Stochastic Depth) per sample.

Parameters:: drop_prob (float, optional, default=0.0) – Probability of dropping paths.

Methods

Attributes

forward(x)[source]¶

Forward pass through DropPath module.

Return type:: Tensor