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

Update finetune_and_eval.md

上级 9859eb33
...@@ -27,14 +27,14 @@ ...@@ -27,14 +27,14 @@
> 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)
> label = fluid.layers.data(name="label", dtype="int64", shape=[1]) > img = input_dict["image"]
> img = input_dict[0] >
> feature_map = output_dict[0] > feature_map = output_dict["feature_map"]
> >
> 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_and_eval( > hub.finetune_and_eval(
> 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.
先完成此消息的编辑!
想要评论请 注册