idiomify / config.yaml
eubinecto's picture
[#9] main_deploy.py: removed authors & versions. config.yaml: incresed epochs with smaller lr. fetchers.py: typo fixed
59df933
# for training an idiomifier
idiomifier:
ver: m-1-3
desc: Just overfitting on PIE dataset, but now with <idiom> & </idiom> special tokens.
bart: facebook/bart-base
lr: 0.00005
literal2idiomatic_ver: d-1-3
idioms_ver: d-1-3
tokenizer_ver: t-1-1
max_epochs: 8
batch_size: 40
shuffle: true
seed: 104
# for building & uploading datasets or tokenizer
idioms:
ver: d-1-3
description: the set of idioms in the traning set of literal2idiomatic_d-1-3. Definitions of them are added as well.
literal2idiomatic:
ver: d-1-3
description: The idioms are annotated with <idiom> & </idiom>.
train_ratio: 0.8
seed: 104
boi_token: <idiom>
eoi_token: </idiom>
tokenizer:
ver: t-1-1
description: A pretrained BartTokenizer. The idiom special tokens are pre-added.
bart: facebook/bart-base
boi_token: <idiom>
eoi_token: </idiom>