mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
* 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
5 lines
182 B
Python
5 lines
182 B
Python
from models.model_factory import create_model
|
|
from models.helpers import load_checkpoint, resume_checkpoint
|
|
from models.test_time_pool import TestTimePoolHead, apply_test_time_pool
|
|
|