diff --git a/model_zoo/alexnet/README.md b/model_zoo/alexnet/README.md index f3333ccabd9f9247a2706a7c88959d969b52542d..1059e22aaeb0d7971ae6a4d8563ab98601e693e1 100644 --- a/model_zoo/alexnet/README.md +++ b/model_zoo/alexnet/README.md @@ -2,7 +2,7 @@ ## Description -Training AlexNet with CIFAR-10 dataset in MindSpore. +Training AlexNet with dataset in MindSpore. This is the simple tutorial for training AlexNet in MindSpore. @@ -10,19 +10,19 @@ This is the simple tutorial for training AlexNet in MindSpore. - Install [MindSpore](https://www.mindspore.cn/install/en). -- Download the CIFAR-10 dataset, the directory structure is as follows: +- Download the dataset, the directory structure is as follows: ``` -├─cifar-10-batches-bin +├─10-batches-bin │ -└─cifar-10-verify-bin +└─10-verify-bin ``` ## Running the example ```python # train AlexNet, hyperparameter setting in config.py -python train.py --data_path cifar-10-batches-bin +python train.py --data_path 10-batches-bin ``` You will get the loss value of each step as following: @@ -38,8 +38,8 @@ epoch: 1 step: 1538, loss is 1.0221305 Then, evaluate AlexNet according to network model ```python -# evaluate AlexNet, 1 epoch training accuracy is up to 51.1%; 10 epoch training accuracy is up to 81.2% -python eval.py --data_path cifar-10-verify-bin --ckpt_path checkpoint_alexnet-1_1562.ckpt +# evaluate AlexNet +python eval.py --data_path 10-verify-bin --ckpt_path checkpoint_alexnet-1_1562.ckpt ``` ## Note diff --git a/model_zoo/alexnet/eval.py b/model_zoo/alexnet/eval.py index 82544de43dbfc8ab350808d70112d0ba1a7820b9..59acd33bc10a0f73c2c3b5f0d101807954beb37d 100644 --- a/model_zoo/alexnet/eval.py +++ b/model_zoo/alexnet/eval.py @@ -19,9 +19,9 @@ python eval.py --data_path /YourDataPath --ckpt_path Your.ckpt """ import argparse -from config import alexnet_cfg as cfg -from dataset import create_dataset_mnist -from alexnet import AlexNet +from src.config import alexnet_cfg as cfg +from src.dataset import create_dataset_mnist +from src.alexnet import AlexNet import mindspore.nn as nn from mindspore import context from mindspore.train.serialization import load_checkpoint, load_param_into_net diff --git a/model_zoo/alexnet/src/__init__.py b/model_zoo/alexnet/src/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/model_zoo/alexnet/alexnet.py b/model_zoo/alexnet/src/alexnet.py similarity index 100% rename from model_zoo/alexnet/alexnet.py rename to model_zoo/alexnet/src/alexnet.py diff --git a/model_zoo/alexnet/config.py b/model_zoo/alexnet/src/config.py similarity index 100% rename from model_zoo/alexnet/config.py rename to model_zoo/alexnet/src/config.py diff --git a/model_zoo/alexnet/dataset.py b/model_zoo/alexnet/src/dataset.py similarity index 100% rename from model_zoo/alexnet/dataset.py rename to model_zoo/alexnet/src/dataset.py diff --git a/model_zoo/alexnet/generator_lr.py b/model_zoo/alexnet/src/generator_lr.py similarity index 100% rename from model_zoo/alexnet/generator_lr.py rename to model_zoo/alexnet/src/generator_lr.py diff --git a/model_zoo/alexnet/train.py b/model_zoo/alexnet/train.py index 2ebadec89f64a3d6128d06dae8199e2c0797a87a..4bac634fe99bfaca4baa78308728b8faf1601829 100644 --- a/model_zoo/alexnet/train.py +++ b/model_zoo/alexnet/train.py @@ -19,10 +19,10 @@ python train.py --data_path /YourDataPath """ import argparse -from config import alexnet_cfg as cfg -from dataset import create_dataset_mnist -from generator_lr import get_lr -from alexnet import AlexNet +from src.config import alexnet_cfg as cfg +from src.dataset import create_dataset_mnist +from src.generator_lr import get_lr +from src.alexnet import AlexNet import mindspore.nn as nn from mindspore import context from mindspore import Tensor diff --git a/model_zoo/lenet/README.md b/model_zoo/lenet/README.md index 750956e74b7d9c6c2fdc6caad6eee0c0a728bc2d..579c9894b2eeca019577a170cef825a29ea68aa0 100644 --- a/model_zoo/lenet/README.md +++ b/model_zoo/lenet/README.md @@ -2,7 +2,7 @@ ## Description -Training LeNet with MNIST dataset in MindSpore. +Training LeNet with dataset in MindSpore. This is the simple and basic tutorial for constructing a network in MindSpore. @@ -10,10 +10,10 @@ This is the simple and basic tutorial for constructing a network in MindSpore. - Install [MindSpore](https://www.mindspore.cn/install/en). -- Download the MNIST dataset, the directory structure is as follows: +- Download the dataset, the directory structure is as follows: ``` -└─MNIST_Data +└─Data ├─test │ t10k-images.idx3-ubyte │ t10k-labels.idx1-ubyte @@ -27,7 +27,7 @@ This is the simple and basic tutorial for constructing a network in MindSpore. ```python # train LeNet, hyperparameter setting in config.py -python train.py --data_path MNIST_Data +python train.py --data_path Data ``` You will get the loss value of each step as following: @@ -43,8 +43,8 @@ epoch: 1 step: 1741, loss is 0.05018193 Then, evaluate LeNet according to network model ```python -# evaluate LeNet, after 1 epoch training, the accuracy is up to 96.5% -python eval.py --data_path MNIST_Data --ckpt_path checkpoint_lenet-1_1875.ckpt +# evaluate LeNet +python eval.py --data_path Data --ckpt_path checkpoint_lenet-1_1875.ckpt ``` ## Note diff --git a/model_zoo/lenet/eval.py b/model_zoo/lenet/eval.py index 445689623f86210ee85d172ec1435a2444f756c6..a9842f4426124d7c3c6dd8aad0fba63bc038624b 100644 --- a/model_zoo/lenet/eval.py +++ b/model_zoo/lenet/eval.py @@ -20,9 +20,9 @@ python eval.py --data_path /YourDataPath --ckpt_path Your.ckpt import os import argparse -from dataset import create_dataset -from config import mnist_cfg as cfg -from lenet import LeNet5 +from src.dataset import create_dataset +from src.config import mnist_cfg as cfg +from src.lenet import LeNet5 import mindspore.nn as nn from mindspore import context from mindspore.train.serialization import load_checkpoint, load_param_into_net @@ -32,10 +32,10 @@ from mindspore.nn.metrics import Accuracy if __name__ == "__main__": - parser = argparse.ArgumentParser(description='MindSpore MNIST Example') + parser = argparse.ArgumentParser(description='MindSpore Lenet Example') parser.add_argument('--device_target', type=str, default="Ascend", choices=['Ascend', 'GPU', 'CPU'], help='device where the code will be implemented (default: Ascend)') - parser.add_argument('--data_path', type=str, default="./MNIST_Data", + parser.add_argument('--data_path', type=str, default="./Data", help='path where the dataset is saved') parser.add_argument('--ckpt_path', type=str, default="", help='if mode is test, must provide\ path where the trained ckpt file') diff --git a/model_zoo/lenet/src/__init__.py b/model_zoo/lenet/src/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/model_zoo/lenet/config.py b/model_zoo/lenet/src/config.py similarity index 100% rename from model_zoo/lenet/config.py rename to model_zoo/lenet/src/config.py diff --git a/model_zoo/lenet/dataset.py b/model_zoo/lenet/src/dataset.py similarity index 100% rename from model_zoo/lenet/dataset.py rename to model_zoo/lenet/src/dataset.py diff --git a/model_zoo/lenet/lenet.py b/model_zoo/lenet/src/lenet.py similarity index 100% rename from model_zoo/lenet/lenet.py rename to model_zoo/lenet/src/lenet.py diff --git a/model_zoo/lenet/train.py b/model_zoo/lenet/train.py index 2c0022be8c4ff0735ae570e50ab5778cd699f89a..80f30310ab1c6ffd8e3a5c652a39546452f3443c 100644 --- a/model_zoo/lenet/train.py +++ b/model_zoo/lenet/train.py @@ -20,9 +20,9 @@ python train.py --data_path /YourDataPath import os import argparse -from config import mnist_cfg as cfg -from dataset import create_dataset -from lenet import LeNet5 +from src.config import mnist_cfg as cfg +from src.dataset import create_dataset +from src.lenet import LeNet5 import mindspore.nn as nn from mindspore import context from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor, TimeMonitor @@ -31,10 +31,10 @@ from mindspore.nn.metrics import Accuracy if __name__ == "__main__": - parser = argparse.ArgumentParser(description='MindSpore MNIST Example') + parser = argparse.ArgumentParser(description='MindSpore Lenet Example') parser.add_argument('--device_target', type=str, default="Ascend", choices=['Ascend', 'GPU', 'CPU'], help='device where the code will be implemented (default: Ascend)') - parser.add_argument('--data_path', type=str, default="./MNIST_Data", + parser.add_argument('--data_path', type=str, default="./Data", help='path where the dataset is saved') parser.add_argument('--dataset_sink_mode', type=bool, default=True, help='dataset_sink_mode is False or True')