Leffa / detectron2 /solver /__init__.py
franciszzj's picture
init code
b213d84
raw
history blame
336 Bytes
# Copyright (c) Facebook, Inc. and its affiliates.
from .build import build_lr_scheduler, build_optimizer, get_default_optimizer_params
from .lr_scheduler import (
LRMultiplier,
LRScheduler,
WarmupCosineLR,
WarmupMultiStepLR,
WarmupParamScheduler,
)
__all__ = [k for k in globals().keys() if not k.startswith("_")]