o rq i@sddlZddlmZmZddlZddlmZddlmZm Z m Z edZ d deej j eej jjfdedefd d Zd d ZedS)N)IterableUnion) ConfigDict)Registrybuild_from_cfg default_groupZ optimizermodelcfg default_argscCsj|duri}t|tjjst|dr)t|jtjjr)t|dr"|j}||d<n||d<t|tt |dS)a build optimizer from optimizer config dict Args: model: A torch.nn.Module or an iterable of parameters. cfg (:obj:`ConfigDict`): config dict for optimizer object. default_args (dict, optional): Default initialization arguments. Nmoduleparams)Z group_keyr ) isinstancetorchnnModulehasattrr parametersr OPTIMIZERSr)rr r rl/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/modelscope/trainers/optimizer/builder.pybuild_optimizer s  rcCsNttjD]\}}|drqt|r$t|tjjr$tj t ||dqdS)N__) module_nameZ module_cls) inspect getmembersrZoptim startswithisclass issubclassZ OptimizerrZregister_moduler)namer rrrregister_torch_optimizers)s r)N)rtypingrrrZmodelscope.utils.configrZmodelscope.utils.registryrrrrrrZ parameter Parameterdictrrrrrrs