add model mae and fintune checkpoint file

mindspore-lab · Aug 7, 2023 · 7b1fe62 · 7b1fe62
1 parent 85450da
commit 7b1fe62
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
@@ -17,6 +17,7 @@
     inceptionv3,
     inceptionv4,
     layers,
+    mae,
     mixnet,
     mlpmixer,
     mnasnet,
@@ -72,6 +73,7 @@
 from .inceptionv3 import *
 from .inceptionv4 import *
 from .layers import *
+from .mae import *
 from .mixnet import *
 from .mlpmixer import *
 from .mnasnet import *
@@ -129,6 +131,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__)