Function register_torch_optimizers

modelscope/trainers/optimizer/builder.py:41–48

()

def register_torch_optimizers():
    for name, module in inspect.getmembers(torch.optim):
        if name.startswith('__'):
            continue
        if inspect.isclass(module) and issubclass(module,
                                                  torch.optim.Optimizer):
            OPTIMIZERS.register_module(
                default_group, module_name=name, module_cls=module)


register_torch_optimizers()