modules/__init__.py

from .models import register, make
from .lr_scheduler import *
from .optimizer import *