idiomify / README.md
eubinecto's picture
[#2] Support for d-1-2 done. Support for m-1-2 partially done (need to implement the testing logic with some valid metrics)
e3c7b5a
|
raw
history blame
637 Bytes
# Idiomify
A human-inspired Idiomifier based on BERT
<img width="807" alt="image" src="https://user-images.githubusercontent.com/56193069/153775460-5ca04edd-e788-442d-b0f1-e780dc0a5724.png">
## Requirements
- wandb
- pytorch-lightning
- transformers
## Versions
### models
format: `m-a-b`
- a: used to indicate a change in the architecture, or a revision of the final product
- b: used to indicate a different version of the same architecture (with a different set of hyperparameters)
### datasets
format: `d-a-b`
- a: used to indicate a change in the dataset we are using
- b: used to indicate a different version of the dataset