| { | |
| "_CHeeSEArguments": "CHeeSEArguments", | |
| "model_name_or_path": "dbmdz/bert-base-german-cased", | |
| "dataset_path": "./dataset.py", | |
| "metric_path": "./metric.py", | |
| "task": "stance_detection", | |
| "first_sentence_inputs": [ | |
| "question" | |
| ], | |
| "second_sentence_inputs": [ | |
| "title", | |
| "snippet", | |
| "paragraphs" | |
| ], | |
| "labels_to_predict": [ | |
| "stance" | |
| ], | |
| "_CHeeSETrainingArguments": "CHeeSETrainingArguments", | |
| "do_train": true, | |
| "do_eval": true, | |
| "do_predict": true, | |
| "do_cross_validation": false, | |
| "output_dir": "/netscratch/schnitzler/bert", | |
| "logging_dir": "/netscratch/schnitzler/bert", | |
| "save_total_limit": 2, | |
| "log_to_file": true, | |
| "logging_strategy": "steps", | |
| "logging_steps": 50, | |
| "cross_validation_folds": 5, | |
| "num_train_epochs": 4, | |
| "per_device_train_batch_size": 8, | |
| "per_device_eval_batch_size": 8, | |
| "fp16": false, | |
| "learning_rate": 3e-05, | |
| "prediction_csv_kwargs": { | |
| "index": false | |
| }, | |
| "prediction_columns_to_include": [ | |
| "title", | |
| "snippet", | |
| "paragraphs", | |
| "question", | |
| "labels", | |
| "stance" | |
| ], | |
| "overwrite_output_dir": true | |
| } |