我们建议创建并激活 conda
环境,以确保在下面安装正确的软件包版本的干净环境。
# Optional but recommended:
conda create -n ithaca python==3.9
conda activate ithaca
克隆此存储库并进入其根目录。通过以下方式安装完整的 ithaca
依赖项(包括训练):
git clone https://github.com/deepmind/ithaca
cd ithaca
pip install --editable .[train]
pip install --upgrade tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple
cd train/
--editable
选项将 ithaca
安装链接到此存储库,以便 import ithaca
将反映对源代码的任何本地修改。
然后,确保您已安装 TensorFlow。如果不这样做,请按照 TensorFlow 网站上的说明安装 CPU 或 GPU 版本。虽然我们使用 Jax 进行训练,但仍然需要 TensorFlow 来加载数据集。
接下来,确保您已将数据集放置在 data/iphi.json
中,请注意单词列表和区域映射也位于该目录中,如果它们在数据集的更新版本中发生更改,则可能需要替换它们。该数据集可以从I.PHI数据集获得。
最后,要运行训练,请运行:
./launch_local.sh
或者,您可以手动运行:
python experiment.py --config=config.py --jaxline_mode=train --logtostderr