add model mae and fintune checkpoint file

mindspore-lab · Sep 5, 2023 · e006a24 · e006a24
1 parent 696f09b
commit e006a24
Show file tree

Hide file tree

Showing 3 changed files with 731 additions and 0 deletions.
diff --git a/mindcv/models/__init__.py b/mindcv/models/__init__.py
@@ -18,6 +18,7 @@
     inceptionv3,
     inceptionv4,
     layers,
+    mae,
     mixnet,
     mlpmixer,
     mnasnet,
@@ -74,6 +75,7 @@
 from .inceptionv3 import *
 from .inceptionv4 import *
 from .layers import *
+from .mae import *
 from .mixnet import *
 from .mlpmixer import *
 from .mnasnet import *
@@ -132,6 +134,7 @@
 __all__.extend(["InceptionV3", "inception_v3"])
 __all__.extend(["InceptionV4", "inception_v4"])
 __all__.extend(layers.__all__)
+__all__.extend(mae.__all__)
 __all__.extend(mixnet.__all__)
 __all__.extend(mlpmixer.__all__)
 __all__.extend(mnasnet.__all__)