pytorch-image-models/scheduler/__init__.py

from scheduler.cosine_lr import CosineLRScheduler
from scheduler.plateau_lr import PlateauLRScheduler
from scheduler.step_lr import StepLRScheduler
from scheduler.tanh_lr import TanhLRScheduler
from scheduler.scheduler_factory import create_scheduler
Uniform pretrained model handling. * All models have 'default_cfgs' dict * load/resume/pretrained helpers factored out * pretrained load operates on state_dict based on default_cfg * test all models in validate * schedule, optim factor factored out * test time pool wrapper applied based on default_cfg 2019-04-11 21:32:16 -07:00			`from scheduler.cosine_lr import CosineLRScheduler`
			`from scheduler.plateau_lr import PlateauLRScheduler`
			`from scheduler.step_lr import StepLRScheduler`
			`from scheduler.tanh_lr import TanhLRScheduler`
			`from scheduler.scheduler_factory import create_scheduler`