如何修改similarity_net中infer模式的模型output?
Created by: bb9696aa
我在使用预训练好的开源pairwise bow similarity_net,目前模型的ouput只有left_bow和pred这两个变量,我想修改模型的返回结果(如令其返回right_bow和pred)。 Readme中提到,可在../models/matching/中定制自己的模型,但我注意到这貌似只适用于training阶段。run_classifier.py中的infer模式甚至都没有调用../models/matching/bow.py,想问一下inference阶段模型的输出如何修改?