str2encoder就是{"fast_text": FastTextEncoder, "bert": BertEncoder}这样一个dict,你放__init__.py里或者recaller.py都行。 train_epoch函数用你注释掉的那个,batchloss先别用。 str2scorer和str2encoder类似,存放的是判断两个向量之间相似度的模块,现在就用cosine函数就行。可以参考下面代码: