From 8fa3707b6c965883046ea8e52c0acd8a5e1daf73 Mon Sep 17 00:00:00 2001 From: Zeyu Chen Date: Mon, 22 Apr 2019 09:58:50 +0800 Subject: [PATCH] Update finetune.md --- docs/API/finetune.md | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/docs/API/finetune.md b/docs/API/finetune.md index 3305d5ab..eacf6f02 100644 --- a/docs/API/finetune.md +++ b/docs/API/finetune.md @@ -27,15 +27,13 @@ > image_width=resnet_module.get_excepted_image_width(), > image_height=resnet_module.get_excepted_image_height(), > dataset=dataset) -> with fluid.program_guard(program): -> label = fluid.layers.data(name="label", dtype="int64", shape=[1]) -> img = input_dict[0] -> feature_map = output_dict[0] +> img = input_dict[0] +> feature_map = output_dict[0] > -> feed_list = [img.name, label.name] +> feed_list = [img.name, label.name] > -> task = hub.create_img_cls_task( -> feature=feature_map, label=label, num_classes=dataset.num_labels) -> hub.finetune( -> task, feed_list=feed_list, data_reader=data_reader) +> task = hub.create_img_cls_task( +> feature=feature_map, num_classes=dataset.num_labels) +> hub.finetune( +> task, feed_list=feed_list, data_reader=data_reader) > ``` -- GitLab