CDial-GPT-tf

CDial-GPT-tf copied to clipboard

本TF版本的输入 `token_ids, segment_ids` 和原版的模型输入 `input_ids, token_type_ids` 是否一样？

9

本TF版predict 输入为 https://github.com/bojone/CDial-GPT-tf/blob/master/example.py#L37 ```bash return model.predict([token_ids, segment_ids])[:, -1] ``` 原版模型输入为 https://github.com/thu-coai/CDial-GPT/blob/master/interact.py#L81 ```bash logits, *_ = model(input_ids, token_type_ids=token_type_ids) ``` 我使用 pytorch 版的输入 `input_ids, token_type_ids` 作为 TF版的输入 `token_ids, segment_ids`，但两个模型的输出结果虽然shape相同，但数值差别很大。 请问什么原因呢？感谢！

wulikai1993