mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
0bc50e84f8
* Move 'test time pool' to Module that can be used by any model, remove from DPN * Remove ResNext model file and combine with ResNet * Remove fbresnet200 as it was an old conversion and pretrained performance not worth param count * Cleanup adaptive avgmax pooling and add back conctat variant * Factor out checkpoint load fn
4 lines
103 B
Python
4 lines
103 B
Python
from .model_factory import create_model, load_checkpoint
|
|
from .test_time_pool import TestTimePoolHead
|
|
|