LightningCLI

LightningCLI#

Attention

在1.0.0版本发布前,当前文档的内容可能会发生变化。

TrajDL是构建在Pytorch Lightning上的算法工具包,因此其设计的每个模型都支持LightningCLI,而且其提供的BaseSeqDataModule是基于LightningModule构建的,其子类都支持LightningModule的特性,完全适配LightningCLI。用户可以通过配置文件的方式快速拉起训练任务,这在批量训练模型的场景里面非常实用。尤其是在实验和生产环境里面。

See also

用户可以查询LightningCLI的官方文档了解配置文件的语法。

TrajDL针对每个SOTA算法提供了YAML配置文件,用户可以通过TrajDL提供的脚本快速运行SOTA模型的实验,所有脚本放置在源代码仓库内的scripts/目录下。