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

Update finetune.md

上级 cb56b6b3
...@@ -27,15 +27,13 @@ ...@@ -27,15 +27,13 @@
> image_width=resnet_module.get_excepted_image_width(), > image_width=resnet_module.get_excepted_image_width(),
> image_height=resnet_module.get_excepted_image_height(), > image_height=resnet_module.get_excepted_image_height(),
> dataset=dataset) > dataset=dataset)
> with fluid.program_guard(program): > img = input_dict[0]
> label = fluid.layers.data(name="label", dtype="int64", shape=[1]) > 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( > task = hub.create_img_cls_task(
> feature=feature_map, label=label, num_classes=dataset.num_labels) > feature=feature_map, num_classes=dataset.num_labels)
> hub.finetune( > hub.finetune(
> task, feed_list=feed_list, data_reader=data_reader) > 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.
先完成此消息的编辑!
想要评论请 注册