文件名称:ai2:使用AI2排行榜测试模型的框架
文件大小:7KB
文件格式:ZIP
更新时间:2024-05-23 09:18:56
Python
AI2常识排行榜的最小代码库 依存关系 如果要使用半精度,安装apex: : 。 还包括Conda env文件以供参考,顶点可能与conda不直接兼容,因此您可以在创建环境之前将其删除。 pip install -r requirements.txt 火车 根据需要修改config.yaml并运行python train.py训练模型。 它加载配置文件并输出所有的日志/检查点outputs 评估 无需评估即可获得预测 python eval.py \ --input_x cache/physicaliqa-train-dev/physicaliqa-train-dev/dev.jsonl \ --config config.yaml \ --checkpoint outputs/2020-02-26/20-26-22/lightning_logs/versio
【文件预览】:
ai2-master
----.gitignore(1KB)
----requirements.txt(166B)
----eval.py(2KB)
----README.md(2KB)
----environment.yml(2KB)
----config.yaml(583B)
----model.py(6KB)
----train.py(1KB)