Spaces:
Sleeping
Sleeping
File size: 874 Bytes
98e2ea5 |
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 |
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'
|