Home
last modified time | relevance | path

Searched defs:TrainingModule (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/executorch/examples/llm_pte_finetuning/
H A Dmodel_loading_lib.py55 model: TrainingModule,
H A Dtraining_lib.py24 class TrainingModule(torch.nn.Module): class
/aosp_15_r20/external/executorch/extension/training/pybindings/
H A D_training_module.py22 class TrainingModule: class
/aosp_15_r20/external/executorch/examples/models/phi-3-mini-lora/
H A Dexport_model.py19 class TrainingModule(torch.nn.Module): class