melbert-roberta / main_config.cfg
EhsanAghazadeh's picture
Add model and its dependencies
e435e5b
raw
history blame contribute delete
425 Bytes
[args]
bert_model=roberta-base
data_dir=VUA20
task_name=vua
model_type=MELBERT
classifier_hidden=768
lr_schedule=warmup_linear
warmup_epoch=2
drop_ratio=0.2
kfold=10
num_bagging=0
bagging_index=0
use_pos=True
use_local_context=True
max_seq_length=150
do_train=True
do_test=True
do_eval=True
do_lower_case=False
class_weight=3
train_batch_size=32
eval_batch_size=8
learning_rate=3e-05
num_train_epoch=3
no_cuda=False
seed=42