huggingface/transfer-learning-conv-ai

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

1754 stars · 432 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

A clean, commented PyTorch codebase for training a dialog/chatbot agent by transfer learning from OpenAI GPT/GPT-2 language models. It reproduces Hugging Face's state-of-the-art ConvAI2 NeurIPS 2018 competition entry with training and interaction scripts plus a pretrained model.

Use cases

• train a chatbot with transfer learning from GPT-2
• reproduce ConvAI2 competition results
• fine-tune GPT-2 for multi-turn dialog
• run a pretrained conversational model locally
• learn how to build a dialog agent in ~250 lines of PyTorch

When to choose

• you want a minimal, readable example of fine-tuning GPT/GPT-2 for conversation
• you need to reproduce the ConvAI2 baseline results
• you want a pretrained dialog model to interact with quickly

When to avoid

• you need a production chatbot framework with integrations and deployment tooling
• you want modern LLM fine-tuning (this predates current tooling and models)
• you need active development or recent model support

Facets

library · maturity maintenance

llm-training chatbot machine-learning nlp large-language-models chatbots deep-learning python gpt-2 transfer-learning dialog-system pytorch convai2 fine-tuning natural-language-processing gpu docker linux

Member repositories

For agents

markdown · JSON · MCP: product_card(name="huggingface/transfer-learning-conv-ai")