From b6ea1d9497a9d4e79304f07eda4df8026c42240f Mon Sep 17 00:00:00 2001 From: yingchen Date: Sat, 31 Oct 2020 16:31:50 +0800 Subject: [PATCH] update save model notebook and readme --- tutorials/notebook/README.md | 1 + tutorials/notebook/save_model.ipynb | 2 ++ 2 files changed, 3 insertions(+) diff --git a/tutorials/notebook/README.md b/tutorials/notebook/README.md index 20142563dd..bd1ecac705 100644 --- a/tutorials/notebook/README.md +++ b/tutorials/notebook/README.md @@ -63,6 +63,7 @@ | 基础使用 | 加载数据集 | [loading_dataset.ipynb](https://gitee.com/mindspore/docs/blob/master/tutorials/notebook/loading_dataset.ipynb) | - 学习MindSpore中加载数据集的方法
- 展示加载常用数据集的方法
- 展示加载MindRecord格式数据集的方法
- 展示加载自定义格式数据集的方法 | 基础使用 | 加载图像数据集 | [loading_image_dataset.ipynb](https://gitee.com/mindspore/docs/blob/master/tutorials/notebook/loading_image_dataset.ipynb) | - 学习加载图像数据集
- 学习处理图像数据集
- 学习增强图像数据集 | 基础使用 | 加载文本数据集 | [loading_text_dataset.ipynb](https://gitee.com/mindspore/docs/blob/master/tutorials/notebook/loading_text_dataset.ipynb) | - 学习加载文本数据集
- 学习处理文本数据集
- 学习文本数据集分词 +| 基础使用         |  保存模型   |     [save_model.ipynb](https://gitee.com/mindspore/docs/blob/master/tutorials/notebook/save_model.ipynb)          | - 了解不同平台用于训练的模型类型
 - 学习如何用不同策略保存训练模型
 - 学习如何将模型导出为不同的文件类型,用于不同平台上的训练 | 处理数据 | 将数据集转换为MindSpore数据格式 | [convert_dataset_to_mindspore_data_format.ipynb](https://gitee.com/mindspore/docs/blob/master/tutorials/notebook/convert_dataset_to_mindspore_data_format/convert_dataset_to_mindspore_data_format.ipynb) | - 展示将MNIST数据集转换为MindSpore数据格式
- 展示将CSV数据集转换为MindSpore数据格式
- 展示将CIFAR-10数据集转换为MindSpore数据格式
- 展示将CIFAR-100数据集转换为MindSpore数据格式
- 展示将ImageNet数据集转换为MindSpore数据格式
- 展示用户自定义生成MindSpore数据格式 | 处理数据 | 数据处理与数据增强 | [data_loading_enhancement.ipynb](https://gitee.com/mindspore/docs/blob/master/tutorials/notebook/data_loading_enhance/data_loading_enhancement.ipynb) | - 学习MindSpore中数据处理和增强的方法
- 展示数据处理、增强方法的实际操作
- 对比展示数据处理前和处理后的效果
- 表述在数据处理、增强后的意义 | 数据处理 | 优化数据准备的性能 | [optimize_the_performance_of_data_preparation.ipynb](https://gitee.com/mindspore/docs/blob/master/tutorials/notebook/optimize_the_performance_of_data_preparation/optimize_the_performance_of_data_preparation.ipynb) | - 数据加载性能优化
- shuffle性能优化
- 数据增强性能优化
- 性能优化方案总结 diff --git a/tutorials/notebook/save_model.ipynb b/tutorials/notebook/save_model.ipynb index 90a8cc8730..a4d3436969 100644 --- a/tutorials/notebook/save_model.ipynb +++ b/tutorials/notebook/save_model.ipynb @@ -134,6 +134,8 @@ "from mindspore.common import dtype as mstype\n", "import matplotlib\n", "import mindspore.dataset as ds\n", + "import os\n", + "os.environ[\"PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION\"]=\"python\"\n", "\n", "\n", "def create_dataset(data_path, batch_size=32, repeat_size=1,\n", -- Gitee