Spaces:
Sleeping
Sleeping
| metrics: ['MUC', 'Bcub', 'CEAFE'] | |
| keep_singletons: True | |
| seed: 45 | |
| train: True | |
| use_wandb: True | |
| desc: "Major Entity Tracking" | |
| override_encoder: False | |
| log_vals: False | |
| # Useful for testing models with different memory architecture than the one trained on | |
| override_memory: False | |
| log_dir_add: "" | |
| device: "cuda:0" | |
| key: "" | |
| defaults: | |
| - _self_ | |
| - datasets: litbank | |
| - model: model | |
| - optimizer: adam | |
| - trainer: train | |
| - infra: local | |
| - experiment: debug | |
| paths: | |
| resource_dir: "../data/" | |
| base_data_dir: ${paths.resource_dir}/raw_data | |
| conll_scorer: ${paths.resource_dir}/reference-coreference-scorers/scorer.pl | |
| base_model_dir: ${infra.work_dir}/../models ## remove /../ | |
| model_dir: null | |
| best_model_dir: null | |
| model_filename: 'model.pth' | |
| model_name: null | |
| model_name_prefix: 'met_' | |
| model_path: null | |
| best_model_path: null | |
| doc_encoder_dirname: 'doc_encoder' | |