/demo_embedding

Primary LanguageJupyter Notebook

Embedding model based on transformer

Ref: https://github.com/karpathy/build-nanogpt/blob/01be6b358941cb1c7561a56353423eba1cc7fe80/train_gpt2.py

poetry run python -m demo_embedding.training 

Colab: https://colab.research.google.com/drive/1IUrbHn6v5F7-t9pL19VegJSx-GjqBESp?usp=sharing

Feature extractions

Links