finetune

收录理由

这个库把 BERT、RoBERTa、GPT、GPT-2、DistilBERT 以及几个卷积序列模型的 TensorFlow 实现，统一封装成一套类似 scikit-learn 的接口。你不需要自己写训练循环，只要实例化一个 Classifier，对数据调用 fit()，微调、保存和重新加载都由库来处理。它还支持两阶段工作流：先在大规模无标注语料上微调，再在较小的有标注数据集上继续训练，这在标注数据稀缺时很实用。Docker 配置让 GPU 和 CPU 环境下的运行都变得简单，估计器接口覆盖了分类、回归和序列标注任务。需要坦白的一点是：它面向的是 TensorFlow 1.x 时代，所以更适合遗留管道，或者作为 Hugging Face 生态兴起之前微调工作方式的参考实现。