tnt

收录理由

这个库出自 PyTorch 官方团队，专为那些每次实验都要重写训练循环样板代码的人准备。它把 fit 和 eval 的循环框架、基于钩子的回调机制打包好，还附带数据加载辅助工具，以及检查点、学习率调度、日志和性能分析等实用功能。它不会强迫你接受一套全新的体系，而是让你保留自己的模型和训练代码，按需挑选需要的组件。无论是想在多个项目间保持统一训练流程的团队，还是搭建自定义流水线的研究者，都可以一点点采用它，因为它与原生 PyTorch 协同工作，而不是取而代之。