diff --git a/model_zoo/README.md b/model_zoo/README.md index 9b2443884b..2ff827928c 100644 --- a/model_zoo/README.md +++ b/model_zoo/README.md @@ -148,7 +148,7 @@ In order to facilitate developers to enjoy the benefits of MindSpore framework a | Loss | 0.108 | | Params (M) | 61.10 | | Checkpoint for Fine tuning | 445MB(.ckpt file) | -| Scripts | https://gitee.com/mindspore/mindspore/tree/master/model_zoo/alexnet| +| Scripts | https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/alexnet| #### [LeNet](#table-of-contents) @@ -168,7 +168,7 @@ In order to facilitate developers to enjoy the benefits of MindSpore framework a | Loss | 0.004 | | Params (M) | 0.06 | | Checkpoint for Fine tuning | 483KB(.ckpt file) | -| Scripts | https://gitee.com/mindspore/mindspore/tree/master/model_zoo/lenet| +| Scripts | https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/lenet| ### Object Detection and Segmentation diff --git a/model_zoo/alexnet/README.md b/model_zoo/official/cv/alexnet/README.md similarity index 100% rename from model_zoo/alexnet/README.md rename to model_zoo/official/cv/alexnet/README.md diff --git a/model_zoo/alexnet/eval.py b/model_zoo/official/cv/alexnet/eval.py similarity index 100% rename from model_zoo/alexnet/eval.py rename to model_zoo/official/cv/alexnet/eval.py diff --git a/model_zoo/alexnet/src/__init__.py b/model_zoo/official/cv/alexnet/src/__init__.py similarity index 100% rename from model_zoo/alexnet/src/__init__.py rename to model_zoo/official/cv/alexnet/src/__init__.py diff --git a/model_zoo/alexnet/src/alexnet.py b/model_zoo/official/cv/alexnet/src/alexnet.py similarity index 98% rename from model_zoo/alexnet/src/alexnet.py rename to model_zoo/official/cv/alexnet/src/alexnet.py index c528ae39e9..2e333a0ad5 100644 --- a/model_zoo/alexnet/src/alexnet.py +++ b/model_zoo/official/cv/alexnet/src/alexnet.py @@ -29,7 +29,7 @@ def fc_with_initialize(input_channels, out_channels): return nn.Dense(input_channels, out_channels, weight, bias) def weight_variable(): - return TruncatedNormal(0.02) # 0.02 + return TruncatedNormal(0.02) class AlexNet(nn.Cell): diff --git a/model_zoo/alexnet/src/config.py b/model_zoo/official/cv/alexnet/src/config.py similarity index 100% rename from model_zoo/alexnet/src/config.py rename to model_zoo/official/cv/alexnet/src/config.py diff --git a/model_zoo/alexnet/src/dataset.py b/model_zoo/official/cv/alexnet/src/dataset.py similarity index 100% rename from model_zoo/alexnet/src/dataset.py rename to model_zoo/official/cv/alexnet/src/dataset.py diff --git a/model_zoo/alexnet/src/generator_lr.py b/model_zoo/official/cv/alexnet/src/generator_lr.py similarity index 100% rename from model_zoo/alexnet/src/generator_lr.py rename to model_zoo/official/cv/alexnet/src/generator_lr.py diff --git a/model_zoo/alexnet/train.py b/model_zoo/official/cv/alexnet/train.py similarity index 100% rename from model_zoo/alexnet/train.py rename to model_zoo/official/cv/alexnet/train.py diff --git a/model_zoo/lenet/README.md b/model_zoo/official/cv/lenet/README.md similarity index 100% rename from model_zoo/lenet/README.md rename to model_zoo/official/cv/lenet/README.md diff --git a/model_zoo/lenet/eval.py b/model_zoo/official/cv/lenet/eval.py similarity index 100% rename from model_zoo/lenet/eval.py rename to model_zoo/official/cv/lenet/eval.py diff --git a/model_zoo/lenet/src/__init__.py b/model_zoo/official/cv/lenet/src/__init__.py similarity index 100% rename from model_zoo/lenet/src/__init__.py rename to model_zoo/official/cv/lenet/src/__init__.py diff --git a/model_zoo/lenet/src/config.py b/model_zoo/official/cv/lenet/src/config.py similarity index 100% rename from model_zoo/lenet/src/config.py rename to model_zoo/official/cv/lenet/src/config.py diff --git a/model_zoo/lenet/src/dataset.py b/model_zoo/official/cv/lenet/src/dataset.py similarity index 100% rename from model_zoo/lenet/src/dataset.py rename to model_zoo/official/cv/lenet/src/dataset.py diff --git a/model_zoo/lenet/src/lenet.py b/model_zoo/official/cv/lenet/src/lenet.py similarity index 100% rename from model_zoo/lenet/src/lenet.py rename to model_zoo/official/cv/lenet/src/lenet.py diff --git a/model_zoo/lenet/train.py b/model_zoo/official/cv/lenet/train.py similarity index 100% rename from model_zoo/lenet/train.py rename to model_zoo/official/cv/lenet/train.py