optim_factory.py 429 B

1234567
  1. # lots of uses of these functions directly, ala 'import timm.optim.optim_factory as optim_factory', fun :/
  2. from ._optim_factory import create_optimizer, create_optimizer_v2, optimizer_kwargs
  3. from ._param_groups import param_groups_layer_decay, param_groups_weight_decay, group_parameters, _layer_map, _group
  4. import warnings
  5. warnings.warn(f"Importing from {__name__} is deprecated, please import via timm.optim", FutureWarning)