Skip to content

Commit

Permalink
add auxp config
Browse files Browse the repository at this point in the history
  • Loading branch information
markus583 committed Jan 12, 2024
1 parent c29f6a9 commit 1fd7730
Showing 1 changed file with 45 additions and 0 deletions.
45 changes: 45 additions & 0 deletions configs/xlmr_stratify_0.1_3layers_p_v2_auxp0.3.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
{
"model_name_or_path": "xlm-roberta-base",
"output_dir": "xlmr-normal-p-v2_auxp0.3",
"train_text_path": "data/sentence/train.parquet",
"valid_text_path": "data/sentence/valid.parquet",
"block_size": 512,
"use_bert": true,
"do_train": true,
"do_eval": true,
"evaluation_strategy": "steps",
"per_device_train_batch_size": 32,
"per_device_eval_batch_size": 32,
"gradient_accumulation_steps": 2,
"eval_accumulation_steps": 8,
"dataloader_num_workers": 4,
"preprocessing_num_workers": 32,
"learning_rate": 1e-4,
"save_strategy": "steps",
"fp16": false,
"max_steps": 200000,
"save_steps": 100000,
"eval_steps": 5000,
"logging_steps": 50,
"report_to": "wandb",
"is_decoder": false,
"remove_unused_columns": false,
"lookahead": null,
"one_sample_per_line": false,
"do_sentence_training": true,
"do_auxiliary_training": true,
"aux_training_weight": 1.0,
"auxiliary_remove_prob": 0.3,
"warmup_steps": 5000,
"adapter_warmup_steps": 0,
"adapter_lr_multiplier": 1,
"ngram_order": 1,
"non_punctuation_sample_ratio": 0.1,
"prediction_loss_only": true,
"use_auxiliary": true,
"ddp_timeout": 3600,
"use_subwords": true,
"num_hidden_layers": 3,
"custom_punctuation_file": "punctuation_xlmr_unk.txt",
"log_level": "warning"
}

0 comments on commit 1fd7730

Please sign in to comment.