← all repositories

920232796/bert_seq2seq

A lightweight PyTorch framework for seq2seq tasks using BERT-style models with UniLM, supporting summarization, classification, NER, and text generation.

1.3k stars Python Language ModelsML Frameworks
bert_seq2seq
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of BERT-based seq2seq using the UniLM approach. It supports multiple pretrained models including BERT, RoBERTa, GPT2, and T5, and enables tasks such as automatic summarization, text classification, sentiment analysis, named entity recognition, and relation extraction. Users configure model type via model_name and task type via model_class, with example scripts provided for each task. A distributed training version supporting multi-GPU training is also available.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.