您好，我正在尝试使用 BERT 问答模型创建一个 .pb 或 .ckpt 文件，以便在转换后将其转换为 tflite 文件，正如官方 tensorflow 文档所说但我还不能得到它，非常感谢

我正在使用 BERT Squad 模型对一组文档 (>20,000) 提出相同的问题。该模型目前在我的 CPU 上运行，处理一个文档大约需要一分钟 - 这意味着我需要几天时间才能完成该程序。

我想知道是否可以通过在 GPU 上运行模型来加快速度。但是，我是 GPU 新手，我不知道如何将这些输入和模型发送到设备 (Titan xp)。

该代码是从 Chris McChormick 那里借来的。

'question' 和 'answer_text' 分别是问题和上下文字符串。

我知道我可以使用 model.tocuda() 将模型发送到 GPU。但是如何发送输入、训练模型以及从 GPU 检索输出？

我有以下查询

1. 数据集格式（是如何拆分训练、测试和有效数据）
2. 放置数据集的位置
3. 如何更改数据集阅读器的路径
4. 如何将模型保存在我自己的目录中
5. 以及如何使用训练好的模型

编辑

我使用此命令更改配置文件中的数据集路径和模型路径。我的模型保存在此位置，但它在训练期间没有使用我的数据集，而是在该文件夹中下载自己的数据集并使用它。

如何使用我的自定义数据集微调预训练的 deeppavlov 小队模型。