Skip to content

This repository contains notebooks and pipelines exploring internal working of GPT2, visualization of its embeddings, its capabilities for different NLP tasks - text completion, classification, reading comprehension etc.

Notifications You must be signed in to change notification settings

Taaniya/exploring-gpt2-language-model

Repository files navigation

exploring-gpt2-language-model

Finetune Transformer decoder only models like GPT2 with causal language model objective

python finetune_gpt2_clm_pytorch.py \
	--model-name gpt2 \
	--dataset-path train.txt \
	--batch-size 8 \
	--epoch 4 \
	--model-version-no 1 \
	--model-dir ./

About

This repository contains notebooks and pipelines exploring internal working of GPT2, visualization of its embeddings, its capabilities for different NLP tasks - text completion, classification, reading comprehension etc.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages