未验证 提交 8fa3707b 编写于 作者: Z Zeyu Chen 提交者: GitHub

Update finetune.md

上级 cb56b6b3
......@@ -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)
> ```
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册