graphgpt/train文件夹下不同的train_[*].py文件的作用？

HKUDS / GraphGPT

[SIGIR'2024] "GraphGPT: Graph Instruction Tuning for Large Language Models"

https://arxiv.org/abs/2310.13023

Apache License 2.0

493 stars 36 forks source link

Open tigerbunny2023 opened 3 months ago

tigerbunny2023 commented 3 months ago

可以看到graphgpt/train文件夹下具有多个不同的train_[*].py：

这些不同文件的作用是什么？

yuh-yang commented 3 months ago

你好，我们主要使用train_graph.py和train_light.py进行训练，后者融合了lightning框架，训练、部署和推理时具有更好的灵活性，其他文件主要来自fastchat库本身