diff --git a/config/model/electra-small-nordic.yaml b/config/model/electra-small-nordic.yaml new file mode 100644 index 0000000..ec5d625 --- /dev/null +++ b/config/model/electra-small-nordic.yaml @@ -0,0 +1,4 @@ +input_model_id: jonfd/electra-small-nordic +output_model_id: alexandrainst/electra-small-nordic-nli-scandi +wandb_run_name: electra-small-nordic +batch_size: 32 diff --git a/config/model/mdeberta_v3_base.yaml b/config/model/mdeberta-v3-base.yaml similarity index 100% rename from config/model/mdeberta_v3_base.yaml rename to config/model/mdeberta-v3-base.yaml diff --git a/config/model/nb_bert_base.yaml b/config/model/nb-bert-base.yaml similarity index 100% rename from config/model/nb_bert_base.yaml rename to config/model/nb-bert-base.yaml diff --git a/config/model/nb_bert_large.yaml b/config/model/nb-bert-large.yaml similarity index 100% rename from config/model/nb_bert_large.yaml rename to config/model/nb-bert-large.yaml diff --git a/config/model/xlm-roberta-base.yaml b/config/model/xlm-roberta-base.yaml index d3226c9..0f8372b 100644 --- a/config/model/xlm-roberta-base.yaml +++ b/config/model/xlm-roberta-base.yaml @@ -1,4 +1,4 @@ input_model_id: xlm-roberta-base output_model_id: alexandrainst/xlm-roberta-base-nli-scandi wandb_run_name: xlm-roberta-base -batch_size: 16 +batch_size: 8