Ross Wightman d9a6a9d0af
Merge pull request #74 from rwightman/augmix-jsd
AugMix, JSD loss, SplitBatchNorm (Auxiliary BN), and more
2020-01-11 12:04:29 -08:00

25 lines
679 B
Python

from .inception_v4 import *
from .inception_resnet_v2 import *
from .densenet import *
from .resnet import *
from .dpn import *
from .senet import *
from .xception import *
from .nasnet import *
from .pnasnet import *
from .selecsls import *
from .efficientnet import *
from .mobilenetv3 import *
from .inception_v3 import *
from .gluon_resnet import *
from .gluon_xception import *
from .res2net import *
from .dla import *
from .hrnet import *
from .registry import *
from .factory import create_model
from .helpers import load_checkpoint, resume_checkpoint
from .test_time_pool import TestTimePoolHead, apply_test_time_pool
from .split_batchnorm import convert_splitbn_model