adas-roberta-base / config.yaml
bluetwigs25's picture
Upload folder using huggingface_hub
ca3e2ca verified
Raw History Blame Contribute Delete
557 Bytes
add_speaker_info: true
ctx_size: 2
data_dir: data/processed/
dropout: 0.1
eval_batch_size: 8
learning_rate: 1.0e-05
log_interval: 10
lstm_hidden_size: 256
lstm_num_layers: 1
max_grad_norm: 1.0
max_length: 64
model_type: simple
num_epochs: 6
pseudo_data_dir: results/pseudo_labeled/CCMS-YearIII
pseudo_label_ratio: 1.5
save_dir: data/results/models/
seed: 42
synthetic_data_ratio: 2.0
synthetic_datafile: data/processed/synthetic_data.csv
train_batch_size: 8
transformer_name: FacebookAI/roberta-base
use_class_weights: true
use_crf: false
warmup_steps: 200