From 0d11c28375f1b5b54e9bf4f7fcea4a255c7061a1 Mon Sep 17 00:00:00 2001 From: changzherui Date: Tue, 22 Sep 2020 10:53:39 +0800 Subject: [PATCH] modify callback --- .../advanced_use/customized_debugging_information.md | 4 ++-- .../source_en/use/saving_and_loading_model_parameters.md | 2 +- .../advanced_use/customized_debugging_information.md | 4 ++-- .../source_zh_cn/use/saving_and_loading_model_parameters.md | 2 +- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tutorials/source_en/advanced_use/customized_debugging_information.md b/tutorials/source_en/advanced_use/customized_debugging_information.md index 9df41eacee..0ad1636505 100644 --- a/tutorials/source_en/advanced_use/customized_debugging_information.md +++ b/tutorials/source_en/advanced_use/customized_debugging_information.md @@ -118,8 +118,8 @@ Here are two examples to further understand the usage of custom Callback. epoch_num = cb_params.cur_epoch_num step_num = cb_params.cur_step_num loss = cb_params.net_outputs - cur_time = time.time() - if (cur_time - cb_params.init_time) > self.run_time: + cur_time = time.time() + if (cur_time - cb_params.init_time) > self.run_time: print("epoch: ", epoch_num, " step: ", step_num, " loss: ", loss) run_context.request_stop() diff --git a/tutorials/source_en/use/saving_and_loading_model_parameters.md b/tutorials/source_en/use/saving_and_loading_model_parameters.md index e0bf53f682..4612a1dbf9 100644 --- a/tutorials/source_en/use/saving_and_loading_model_parameters.md +++ b/tutorials/source_en/use/saving_and_loading_model_parameters.md @@ -117,7 +117,7 @@ resnet = ResNet50() load_checkpoint("resnet50-2_32.ckpt", net=resnet) dateset_eval = create_dataset(os.path.join(mnist_path, "test"), 32, 1) # define the test dataset loss = CrossEntropyLoss() -model = Model(resnet, loss) +model = Model(resnet, loss, metrics={"Accuracy": Accuracy()}) acc = model.eval(dataset_eval) ``` diff --git a/tutorials/source_zh_cn/advanced_use/customized_debugging_information.md b/tutorials/source_zh_cn/advanced_use/customized_debugging_information.md index 42c1eec0e9..957162c885 100644 --- a/tutorials/source_zh_cn/advanced_use/customized_debugging_information.md +++ b/tutorials/source_zh_cn/advanced_use/customized_debugging_information.md @@ -120,8 +120,8 @@ class Callback(): epoch_num = cb_params.cur_epoch_num step_num = cb_params.cur_step_num loss = cb_params.net_outputs - cur_time = time.time() - if (cur_time - cb_params.init_time) > self.run_time: + cur_time = time.time() + if (cur_time - cb_params.init_time) > self.run_time: print("epoch: ", epoch_num, " step: ", step_num, " loss: ", loss) run_context.request_stop() diff --git a/tutorials/source_zh_cn/use/saving_and_loading_model_parameters.md b/tutorials/source_zh_cn/use/saving_and_loading_model_parameters.md index 327040e973..18b297605a 100644 --- a/tutorials/source_zh_cn/use/saving_and_loading_model_parameters.md +++ b/tutorials/source_zh_cn/use/saving_and_loading_model_parameters.md @@ -118,7 +118,7 @@ resnet = ResNet50() load_checkpoint("resnet50-2_32.ckpt", net=resnet) dateset_eval = create_dataset(os.path.join(mnist_path, "test"), 32, 1) # define the test dataset loss = CrossEntropyLoss() -model = Model(resnet, loss) +model = Model(resnet, loss, metrics={"Accuracy": Accuracy()}) acc = model.eval(dataset_eval) ``` -- Gitee