From 8924836f774432027af6ad8f49b3646642ed06a2 Mon Sep 17 00:00:00 2001 From: Zeyu Chen Date: Mon, 22 Apr 2019 10:03:45 +0800 Subject: [PATCH] Update finetune_and_eval.md --- docs/API/finetune_and_eval.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docs/API/finetune_and_eval.md b/docs/API/finetune_and_eval.md index ee878075..fafc55e8 100644 --- a/docs/API/finetune_and_eval.md +++ b/docs/API/finetune_and_eval.md @@ -27,6 +27,7 @@ > image_width=resnet_module.get_excepted_image_width(), > image_height=resnet_module.get_excepted_image_height(), > dataset=dataset) + > img = input_dict["image"] > > feature_map = output_dict["feature_map"] @@ -36,5 +37,5 @@ > task = hub.create_img_cls_task( > feature=feature_map, num_classes=dataset.num_labels) > hub.finetune_and_eval( -> task, feed_list=feed_list, data_reader=data_reader) +> task=task, feed_list=feed_list, data_reader=data_reader) > ``` -- GitLab