finetune后的模型进行predict时候,需要重新实例task,num_classes参数还需要传入dataset?
Created by: zhangmianhongni
finetune后的模型进行predict时候,需要重新实例task,num_classes参数还需要传入finetune时候的dataset吗?模型保存的时候不是已经保存了相关的label吗?
task = hub.ImageClassifierTask( data_reader=reader, feature=feature_map, feed_list=feed_list, num_classes=dataset.num_labels, metrics_choices=metrics_choices, config=self.config)