| { | |
| "monitor_metric": "mean_f1", | |
| "trainer_kwargs": { | |
| "class_name": "Event_Extractor_Code" | |
| }, | |
| "trainee_kwargs": { | |
| "class_name": "EE", | |
| "prompt_cls": "event_extraction_code", | |
| "loss": { | |
| "class_name": "NLLLoss" | |
| }, | |
| "freeze_prefixes": [] | |
| }, | |
| "data_module_kwargs": { | |
| "class_name": "event_extraction_DataModule", | |
| "data_processor": { | |
| "class_name": "viquae_data_processor", | |
| "train_path": "viquae/train/", | |
| "val_path": "viquae/val/", | |
| "test_path": "viquae/test/" | |
| }, | |
| "dataloader_kwargs": { | |
| "num_workers": 6, | |
| "prefetch_factor": 2 | |
| } | |
| }, | |
| "answer_generator_kwargs": { | |
| "dividable": true, | |
| "checkpoint_name": "answer_generator", | |
| "rag_training": false, | |
| "rag": false, | |
| "wo_kb": true, | |
| "generator_only": true, | |
| "entity_prompt": false, | |
| "beam_search": true, | |
| "best_answer": false, | |
| "best_entity": false, | |
| "inference_path": "saved_models/wikievents/GoLLIE-7B/code_generator/train/2025-09-26/19-34-03/answer_generator" | |
| }, | |
| "all_datasets_path": "../datasets" | |
| } |