6 contributors

History: 22 commits

versae

Adding base config and organizing configs

9c5541b over 3 years ago

configs
Adding base config and organizing configs over 3 years ago
mc4
Adding sampling to mc4 over 3 years ago
.gitattributes

690 Bytes

initial commit over 3 years ago
.gitignore

1.84 kB

Initial test with BETO's corpus over 3 years ago
README.md

1.84 kB

Adding checkpointing, wandb, and new mlm script over 3 years ago
config.py

247 Bytes

Scripts for perplexity sampling and fixes over 3 years ago
convert.py

458 Bytes

Scripts for perplexity sampling and fixes over 3 years ago
flax_model.msgpack

712 MB
LFS

Adding correct models 10k steps over 3 years ago
get_embeddings_and_perplexity.py

1.53 kB

Add script to generate dataset of embeddings and perplexities. Add script to generate t-SNE plot for embedding and perplexity visualization. over 3 years ago
perplexity.py

751 Bytes

Adding checkpointing, wandb, and new mlm script over 3 years ago
pytorch_model.bin

1.42 GB
LFS

Adding correct models 10k steps over 3 years ago
run.sh

883 Bytes

Adding base config and organizing configs over 3 years ago
run_mlm_flax.py

30 kB

Adding sampling to mc4 over 3 years ago
run_mlm_flax_stream.py

28 kB

Adding sampling to mc4 over 3 years ago
run_stream.sh

924 Bytes

Adding base config and organizing configs over 3 years ago
tokenizer.json

1.47 MB

New tokenizer over 3 years ago
tokens.py

649 Bytes

Scripts for perplexity sampling and fixes over 3 years ago
tokens.py.orig

899 Bytes

Adjust batch size for extrating tokens over 3 years ago
tsne_plot.py

3.02 kB

Remove unused imports over 3 years ago