fl4health.model_bases.masked_layers.masked_layers_utils module

convert_to_masked_model(original_model)[source]

Given a model, convert every one of its layers to a masked layer of the same kind, if applicable.

Return type:

Module

is_masked_module(module)[source]
Return type:

bool