Searched defs:extract_weights (Results 1 – 3 of 3) sorted by relevance
47 def extract_weights(mod: nn.Module) -> Tuple[Tuple[Tensor, ...], List[str]]: function
96 def extract_weights( function
382 def extract_weights( function