Skip to content

marlcplhra/ShuffleNet

Repository files navigation

ShuffleNet

ShuffleNet-v1 based on MindSpore

有关模型的介绍请参见"ShuffleNet-v1.ipynb"(训练过程代码请以以下脚本为准).

训练脚本:

预训练脚本python ./train.py --train_dataset_path ./data/cifar10
预训练评估脚本python ./eval.py --eval_dataset_path ./data/cifar10 --ckpt_path [ckpt]
迁移训练脚本python ./train.py --config_path=./transfer_config.yaml --resume=[ckpt] --dataset_path=./data/flower_photos
迁移训练评估脚本python ./eval.py --config_path=./transfer_config.yaml --dataset_path=./data/flower_photos

在ckpt文件夹下保存了预训练完成的模型shufflenetv1-250_195.ckpt和迁移训练完成的模型shufflenetv1_transfer_best.ckpt

About

ShuffleNet-v1 based on MindSpore

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published