Searched defs:load_weights (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/ |
| H A D | modeling_base.py | 33 def load_weights(self, state_dict, state_dict_start_idx, verbose): member in BaseModelChunk
|
| H A D | modeling_common.py | 358 def load_weights(self, state_dict, state_dict_start_idx): member in ModelChunk
|
| /aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/ |
| H A D | utils.py | 68 def load_weights(mod: nn.Module, names: List[str], params: Tuple[Tensor, ...]) -> None: function
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | make_functional.py | 116 def load_weights( function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training_v1.py | 181 def load_weights(self, filepath, by_name=False, skip_mismatch=False): member in Model
2866 def load_weights(self, filepath, by_name=False): member in DistributedCallbackModel
|
| H A D | training.py | 2252 def load_weights(self, member in Model
|