fast-reid/fastreid/solver/optim/__init__.py

4 lines
70 B
Python

from .lamb import Lamb
from .swa import SWA
from torch.optim import *