has_dropout_layers

tiatoolbox . models . engine . mc_dropout . has_dropout_layers

has_dropout_layers(model)

Check if the model has any dropout layers.

Parameters:

model (nn.Module) – The model to inspect.

Returns:

True if at least one submodule is an instance of a Dropout type.

Return type:

bool