From 1c37b1a6accb4f12781e2856e09d417eab0fd33d Mon Sep 17 00:00:00 2001
From: zhangyihui <zhangyihui7@huawei.com>
Date: Sat, 26 Dec 2020 15:17:57 +0800
Subject: [PATCH] fix bug

---
 .../advanced_use/performance_profiling.md     | 35 +++++++------------
 .../advanced_use/performance_profiling.md     | 35 +++++++------------
 2 files changed, 24 insertions(+), 46 deletions(-)

diff --git a/tutorials/training/source_en/advanced_use/performance_profiling.md b/tutorials/training/source_en/advanced_use/performance_profiling.md
index 4d9242608d..4fbc78b8f8 100644
--- a/tutorials/training/source_en/advanced_use/performance_profiling.md
+++ b/tutorials/training/source_en/advanced_use/performance_profiling.md
@@ -44,30 +44,19 @@ The sample code is as follows:
 from mindspore.profiler import Profiler
 from mindspore import Model, nn, context
 
+# Init context env
+context.set_context(mode=context.GRAPH_MODE, device_target='Ascend', device_id=int(os.environ["DEVICE_ID"]))
 
-def test_profiler():
-    # Init context env
-    context.set_context(mode=context.GRAPH_MODE, device_target='Ascend', device_id=int(os.environ["DEVICE_ID"]))
-
-    # Init Profiler
-    # Note that 'data' directory is created in current path by default. To visualize the profiling data by MindInsight,
-    # 'data' directory should be placed under summary-base-dir.
-    profiler = Profiler()
-
-    # Init hyperparameter
-    epoch = 2
-    # Init network and Model
-    net = Net()
-    loss_fn = CrossEntropyLoss()
-    optim = MyOptimizer(learning_rate=0.01, params=network.trainable_params())
-    model = Model(net, loss_fn=loss_fn, optimizer=optim, metrics=None)  
-    # Prepare mindrecord_dataset for training
-    train_ds = create_mindrecord_dataset_for_training()
-    # Model Train
-    model.train(epoch, train_ds)
-
-    # Profiler end
-    profiler.analyse()
+# Init Profiler
+# Note that 'data' directory is created in current path by default. To visualize the profiling data by MindInsight,
+# 'data' directory should be placed under summary-base-dir.
+profiler = Profiler()
+
+# Model Train
+Model.train()
+
+# Profiler end
+profiler.analyse()
 ```
 
 ## Launch MindInsight
diff --git a/tutorials/training/source_zh_cn/advanced_use/performance_profiling.md b/tutorials/training/source_zh_cn/advanced_use/performance_profiling.md
index 3a53f103ad..fec0808fab 100644
--- a/tutorials/training/source_zh_cn/advanced_use/performance_profiling.md
+++ b/tutorials/training/source_zh_cn/advanced_use/performance_profiling.md
@@ -48,30 +48,19 @@
 from mindspore.profiler import Profiler
 from mindspore import Model, nn, context
 
+# Init context env
+context.set_context(mode=context.GRAPH_MODE, device_target='Ascend', device_id=int(os.environ["DEVICE_ID"]))
 
-def test_profiler():
-    # Init context env
-    context.set_context(mode=context.GRAPH_MODE, device_target='Ascend', device_id=int(os.environ["DEVICE_ID"]))
-
-    # Init Profiler
-    # Note that 'data' directory is created in current path by default. To visualize the profiling data by MindInsight,
-    # 'data' directory should be placed under summary-base-dir.
-    profiler = Profiler()
-
-    # Init hyperparameter
-    epoch = 2
-    # Init network and Model
-    net = Net()
-    loss_fn = CrossEntropyLoss()
-    optim = MyOptimizer(learning_rate=0.01, params=network.trainable_params())
-    model = Model(net, loss_fn=loss_fn, optimizer=optim, metrics=None)  
-    # Prepare mindrecord_dataset for training
-    train_ds = create_mindrecord_dataset_for_training()
-    # Model Train
-    model.train(epoch, train_ds)
-
-    # Profiler end
-    profiler.analyse()
+# Init Profiler
+# Note that 'data' directory is created in current path by default. To visualize the profiling data by MindInsight,
+# 'data' directory should be placed under summary-base-dir.
+profiler = Profiler()
+
+# Model Train
+Model.train()
+
+# Profiler end
+profiler.analyse()
 ```
 
 ## 启动MindInsight
-- 
Gitee