aste_stage_ver3 / config.json
lmartinson's picture
Update config.json
8210158 verified
{
"class_n": 6,
"dropout": 0.5,
"gcn_dropout": 0.5,
"gcn_num_layers": 3,
"hidden_dim": 64,
"pretrained_model_path": "ai-forever/ruBert-base",
"span_average": false,
"adam_epsilon": 1e-8,
"add_pos_enc": true,
"batch_size": 1,
"bert_lr": 2e-5,
"class_n": 6,
"device": "cpu",
"do_lower_case": true,
"dropout_rate": 0.5,
"hidden_dim": 64,
"init_vocab": "ai-forever/ruBert-base",
"l2": 0.0,
"lr": 1e-4,
"max_len": 1408,
"max_span_length": 4,
"Only_token_head": false,
"order_input": false,
"pretrained_model": "ai-forever/ruBert-base",
"random_shuffle": false,
"span_average": false,
"span_generation": "Max",
"train_batch_size": 1,
"version": "1D",
"with_weight": true
}