GLiNER
/
config.yaml
38 строк · 884.0 Байт
1# Model Configuration
2model_name: microsoft/deberta-v3-small # Hugging Face model
3name: "large"
4max_width: 12
5hidden_size: 768
6dropout: 0.4
7fine_tune: true
8subtoken_pooling: first
9span_mode: markerV0
10
11# Training Parameters
12num_steps: 30000
13train_batch_size: 3
14eval_every: 5000
15warmup_ratio: 0.1
16
17# Learning Rate Configuration
18lr_encoder: 1e-5
19lr_others: 5e-5
20
21# Directory Paths
22root_dir: gliner_logs
23train_data: "data.json" # see https://github.com/urchade/GLiNER/tree/main/data
24val_data_dir: "none"
25# "NER_datasets": val data from the paper can be obtained from "https://drive.google.com/file/d/1T-5IbocGka35I7X3CE6yKe5N_Xg2lVKT/view"
26
27# Pretrained Model Path
28# Use "none" if no pretrained model is being used
29prev_path: "none"
30
31# Advanced Training Settings
32size_sup: -1
33max_types: 25
34shuffle_types: true
35random_drop: true
36max_neg_type_ratio: 1
37max_len: 384
38freeze_token_rep: false
39