File size: 3,638 Bytes
0763f3c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 |
{
"_name_or_path": "allenai/led-base-16384",
"accelerator": null,
"accumulate_grad_batches": 2,
"activation_dropout": 0.0,
"activation_function": "gelu",
"add_context": false,
"amp_backend": null,
"amp_level": null,
"architectures": [
"LEDForConditionalGeneration"
],
"attention_dropout": 0.0,
"attention_window": [
1024,
1024,
1024,
1024,
1024,
1024
],
"auto_lr_find": false,
"auto_scale_batch_size": false,
"auto_select_gpus": null,
"batch_size": 8,
"benchmark": null,
"bos_token_id": 0,
"check_val_every_n_epoch": 1,
"checkpoint": null,
"class_labels": {
"delete": 3,
"ignore": 0,
"rephrase": 1,
"split": 2
},
"classif_dropout": 0.0,
"classifier_dropout": 0.0,
"context_dir": null,
"context_doc_id": null,
"context_window": 13,
"d_model": 768,
"decoder_attention_heads": 12,
"decoder_ffn_dim": 3072,
"decoder_layerdrop": 0.0,
"decoder_layers": 6,
"decoder_start_token_id": 2,
"default_root_dir": null,
"detect_anomaly": false,
"devices": "2",
"dropout": 0.1,
"enable_checkpointing": true,
"enable_model_summary": true,
"enable_progress_bar": true,
"encoder_attention_heads": 12,
"encoder_ffn_dim": 3072,
"encoder_layerdrop": 0.0,
"encoder_layers": 6,
"eos_token_id": 2,
"eval_beams": null,
"eval_max_length": null,
"fast_dev_run": false,
"gpus": null,
"gradient_checkpointing": false,
"gradient_clip_algorithm": null,
"gradient_clip_val": null,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1",
"2": "LABEL_2"
},
"inference_mode": true,
"init_std": 0.02,
"ipus": null,
"is_encoder_decoder": true,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1,
"LABEL_2": 2
},
"limit_predict_batches": null,
"limit_test_batches": null,
"limit_train_batches": null,
"limit_val_batches": null,
"log_every_n_steps": 50,
"logger": true,
"longformer": true,
"lr": 2e-05,
"lr_scheduler": false,
"max_decoder_position_embeddings": 1024,
"max_encoder_position_embeddings": 16384,
"max_epochs": null,
"max_length": 1024,
"max_samples": -1,
"max_steps": -1,
"max_time": null,
"min_epochs": null,
"min_steps": null,
"model_type": "led",
"move_metrics_to_cpu": false,
"multiple_trainloader_mode": "max_size_cycle",
"name": "o-led-para",
"num_hidden_layers": 6,
"num_nodes": 1,
"num_processes": null,
"num_sanity_val_steps": 2,
"op_col": "label",
"op_tokens": [
"<COPY>",
"<REPHRASE>",
"<SPLIT>",
"<DELETE>"
],
"overfit_batches": 0.0,
"pad_token_id": 1,
"plan_col": "labels",
"plan_loss": null,
"plan_prefix": false,
"plan_sep": false,
"plugins": null,
"precision": 32,
"prefix_only": false,
"profiler": null,
"project": "simplification_models",
"reading_lvl": null,
"reload_dataloaders_every_n_epochs": 0,
"replace_sampler_ddp": true,
"resume_from_checkpoint": null,
"save_dir": null,
"sent_level": true,
"simple_context_dir": null,
"simple_context_doc_id": null,
"skip_val_gen": true,
"strategy": null,
"sync_batchnorm": false,
"torch_dtype": "float32",
"tpu_cores": null,
"track_grad_norm": -1,
"train_check_interval": 0.01,
"train_data_dir": null,
"train_file": "data/wikiauto_para_train.csv",
"train_split": 0.9,
"train_workers": 32,
"transformers_version": "4.29.1",
"use_cache": true,
"val_check_interval": null,
"val_file": "data/wikiauto_para_valid.csv",
"val_metric": "bleu",
"val_split": 0.05,
"val_workers": 8,
"valid_data_dir": null,
"vocab_size": 50274,
"wandb_id": null,
"x_col": "complex",
"y_col": "simple"
}
|