mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
fast-reid/fastreid/solver/optim/__init__.py

4 lines
70 B
Python
Raw Normal View History

update version0.2 code
2020-03-25 10:58:26 +08:00
from .lamb import Lamb
feat: add swa algorithm Add swa and related config options, if it is enabled, model will do swa after regular training
2020-05-06 10:17:44 +08:00
from .swa import SWA
v0.3 update Summary: 1. change DPP training in apex way; 2. make warmup scheduler by iter and lr scheduler by epoch; 3. replace random erasing with torchvision implementation; 4. naming modification in config file
2020-12-07 14:19:20 +08:00
from torch.optim import *