ArneBinder commited on
Commit
d2131a6
·
verified ·
1 Parent(s): 708f6f8

fix: upload https://github.com/ArneBinder/pie-document-level/pull/397

Browse files
argumentation_model/_joint.yaml DELETED
@@ -1,4 +0,0 @@
1
- _target_: pytorch_ie.auto.AutoPipeline.from_pretrained
2
- pretrained_model_name_or_path: ???
3
- # this batch_size that works good (fastest) on a single RTX2080Ti (11GB) (see https://github.com/ArneBinder/pie-document-level/issues/334#issuecomment-2613232344)
4
- batch_size: 1
 
 
 
 
 
argumentation_model/_pipelined.yaml DELETED
@@ -1,17 +0,0 @@
1
- _target_: src.pipeline.NerRePipeline
2
- ner_model_path: ???
3
- re_model_path: ???
4
- entity_layer: labeled_spans
5
- relation_layer: binary_relations
6
- # this works good on a single RTX2080Ti (11GB)
7
- ner_pipeline:
8
- batch_size: 256
9
- re_pipeline:
10
- batch_size: 64
11
- # convert the RE model to half precision for mixed precision inference (speedup approx. 4x)
12
- half_precision_model: true
13
- taskmodule_kwargs:
14
- # don't show statistics after encoding
15
- collect_statistics: false
16
- # don't show pipeline steps
17
- verbose: false
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
argumentation_model/joint.yaml DELETED
@@ -1,10 +0,0 @@
1
- defaults:
2
- - _joint
3
-
4
- # best model based on the validation set (see https://github.com/ArneBinder/pie-document-level/issues/334#issuecomment-2613232344 for details)
5
- # i.e. models from https://github.com/ArneBinder/pie-document-level/issues/334#issuecomment-2578422544, but with last checkpoint (instead of best validation checkpoint)
6
- # model_name_or_path: models/dataset-sciarg/task-ner_re/v0.4/2025-01-09_01-50-53
7
- # ckpt_path: logs/training/multiruns/dataset-sciarg/task-ner_re/v0.4/2025-01-09_01-50-52/2/checkpoints/last.ckpt
8
- # w&b run (for the loaded checkpoint): [icy-glitter-5](https://wandb.ai/arne/dataset-sciarg-task-ner_re-v0.4-training/runs/it5toj6w)
9
- pretrained_model_name_or_path: "ArneBinder/sam-pointer-bart-base-v0.4"
10
- revision: "0445c69bafa31f8153aaeafc1767fad84919926a"
 
 
 
 
 
 
 
 
 
 
 
argumentation_model/joint_hps.yaml DELETED
@@ -1,7 +0,0 @@
1
- defaults:
2
- - _joint
3
-
4
- # from: hparams_search for all datasets
5
- # see https://github.com/ArneBinder/pie-document-level/pull/381#issuecomment-2682711151
6
- # THESE ARE LOCAL PATHS, NOT HUGGINGFACE MODELS!
7
- pretrained_model_name_or_path: models/dataset-sciarg/task-ner_re/2025-02-23_05-16-45
 
 
 
 
 
 
 
 
argumentation_model/pipelined.yaml DELETED
@@ -1,8 +0,0 @@
1
- defaults:
2
- - _pipelined
3
-
4
- # from: train pipeline models with bigger train set,
5
- # see https://github.com/ArneBinder/pie-document-level/issues/355#issuecomment-2612958658
6
- # THESE ARE LOCAL PATHS, NOT HUGGINGFACE MODELS!
7
- ner_model_path: models/dataset-sciarg/task-adus/v0.4/2025-01-20_05-50-00
8
- re_model_path: models/dataset-sciarg/task-relations/v0.4/2025-01-22_20-36-23
 
 
 
 
 
 
 
 
 
argumentation_model/pipelined_deprecated.yaml DELETED
@@ -1,9 +0,0 @@
1
- defaults:
2
- - _pipelined
3
-
4
- # from: train pipeline models with bigger train set, but with strange choice of models,
5
- # see edit history of https://github.com/ArneBinder/pie-document-level/issues/355#issuecomment-2612958658
6
- # NOTE: these were originally in the pipelined.yaml
7
- # THESE ARE LOCAL PATHS, NOT HUGGINGFACE MODELS!
8
- ner_model_path: models/dataset-sciarg/task-adus/v0.4/2025-01-20_09-09-11
9
- re_model_path: models/dataset-sciarg/task-relations/v0.4/2025-01-22_12-44-51
 
 
 
 
 
 
 
 
 
 
argumentation_model/pipelined_hps.yaml DELETED
@@ -1,8 +0,0 @@
1
- defaults:
2
- - _pipelined
3
-
4
- # from: hparams_search for all datasets,
5
- # see https://github.com/ArneBinder/pie-document-level/pull/381#issuecomment-2684865102
6
- # THESE ARE LOCAL PATHS, NOT HUGGINGFACE MODELS!
7
- ner_model_path: models/dataset-sciarg/task-adur/2025-02-26_07-14-59
8
- re_model_path: models/dataset-sciarg/task-are/2025-02-20_18-09-25
 
 
 
 
 
 
 
 
 
argumentation_model/pipelined_new.yaml DELETED
@@ -1,14 +0,0 @@
1
- defaults:
2
- - _pipelined
3
-
4
- # from: Update scientific ARE experiment configs,
5
- # see https://github.com/ArneBinder/pie-document-level/pull/379#issuecomment-2651669398
6
- # i.e. the models are now on Hugging Face
7
- # ner_model_path: models/dataset-sciarg/task-adur/2025-02-09_23-08-37
8
- # re_model_path: models/dataset-sciarg/task-are/2025-02-10_19-24-52
9
- ner_model_path: ArneBinder/sam-adur-sciarg
10
- ner_pipeline:
11
- revision: bcbef4e585a5f637009ff702661cf824abede6b0
12
- re_model_path: ArneBinder/sam-are-sciarg
13
- re_pipeline:
14
- revision: 93024388330c58daf20963c2020e08f54553e74c