knowledge-encode / README.md
ruslandev's picture
Create README.md
54ed95e verified
|
Raw
History Blame Contribute Delete
634 Bytes

Knowledge encoding by examples of Word2Vec and LLM training

This repository contains weights for a list of language models:

  • word2vec.pt: embedding trained on 150mil pairs of text tokens subsampled from text8 dataset. SkipGram method with negative sampling was used as described in the original paper.
  • mlp.pt: 2-layers MLP trained on the same dataset and using pretrained embeddings.
  • mlp_norm.pt: Version of the MLP model utilizing LayerNorm for better scaling of the learned features distribution.

Training code can be found on GitHub.