mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
274a67223e
* [Feature] Implement layer-wise learning rate decay optimizer constructor. * Use num_layers instead of max_depth to avoid misleading * Add UT * Update docstring * Update log info * update LearningRateDecay configs --------- Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
9 lines
297 B
Python
9 lines
297 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .adan_t import Adan
|
|
from .lamb import Lamb
|
|
from .lars import LARS
|
|
from .layer_decay_optim_wrapper_constructor import \
|
|
LearningRateDecayOptimWrapperConstructor
|
|
|
|
__all__ = ['Lamb', 'Adan', 'LARS', 'LearningRateDecayOptimWrapperConstructor']
|