adding src/timm/models

microsoft · May 8, 2022 · 46c1627 · 46c1627
1 parent 59934d6
commit 46c1627
Show file tree

Hide file tree

Showing 89 changed files with 23,072 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -39,7 +39,7 @@ tmp*.py
 */*output*/
 linjli
 ./datasets
-models
+./models
 
 
 # compiled files #

diff --git a/src/timm/models/__init__.py b/src/timm/models/__init__.py
@@ -0,0 +1,46 @@
+from .byoanet import *
+from .byobnet import *
+from .cspnet import *
+from .densenet import *
+from .dla import *
+from .dpn import *
+from .efficientnet import *
+from .ghostnet import *
+from .gluon_resnet import *
+from .gluon_xception import *
+from .hardcorenas import *
+from .hrnet import *
+from .inception_resnet_v2 import *
+from .inception_v3 import *
+from .inception_v4 import *
+from .mobilenetv3 import *
+from .nasnet import *
+from .nfnet import *
+from .pit import *
+from .pnasnet import *
+from .regnet import *
+from .res2net import *
+from .resnest import *
+from .resnet import *
+from .resnetv2 import *
+from .rexnet import *
+from .selecsls import *
+from .senet import *
+from .sknet import *
+from .swin_transformer import *
+from .tnt import *
+from .tresnet import *
+from .vgg import *
+from .vision_transformer import *
+from .vision_transformer_hybrid import *
+from .vovnet import *
+from .xception import *
+from .xception_aligned import *
+
+from .factory import create_model, split_model_name, safe_model_name
+from .helpers import load_checkpoint, resume_checkpoint, model_parameters
+from .layers import TestTimePoolHead, apply_test_time_pool
+from .layers import convert_splitbn_model
+from .layers import is_scriptable, is_exportable, set_scriptable, set_exportable, is_no_jit, set_no_jit
+from .registry import register_model, model_entrypoint, list_models, is_model, list_modules, is_model_in_modules,\
+    has_model_default_key, is_model_default_key, get_model_default_value, is_model_pretrained