Ross Wightman 9c3859fb9c 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

4 lines
171 B
Python

from data.dataset import Dataset
from data.transforms import *
from data.loader import create_loader
from data.random_erasing import RandomErasingTorch, RandomErasingNumpy