def is_leaf_module()

in core/train_pipeline.py [0:0]

• 4 lines of code
• 3 McCabe index (conditional complexity)

  def is_leaf_module(self, m: torch.nn.Module, module_qualified_name: str) -> bool:
    if isinstance(m, ShardedModule) or module_qualified_name in self._leaf_modules:
      return True
    return super().is_leaf_module(m, module_qualified_name)