Closed yxwsfz closed 3 years ago
你好，在finetune.py文件中提示缺少self.last_token函数
class AutoTitle(AutoRegressiveDecoder): """seq2seq解码器 """ @AutoRegressiveDecoder.wraps(default_rtype='probas') def predict(self, inputs, output_ids, states): c_encoded = inputs[0] return self.last_token(decoder).predict([c_encoded, output_ids])
你好，在finetune.py文件中提示缺少self.last_token函数
class AutoTitle(AutoRegressiveDecoder): """seq2seq解码器 """ @AutoRegressiveDecoder.wraps(default_rtype='probas') def predict(self, inputs, output_ids, states): c_encoded = inputs[0] return self.last_token(decoder).predict([c_encoded, output_ids])