Searched defs:module_contains_param (Results 1 – 2 of 2) sorted by relevance
7 def module_contains_param(module, parametrization): function
19 def module_contains_param(module: nn.Module, parametrization: Type[nn.Module]) -> bool: function