File size: 851 Bytes
8df825c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
{
"activation": "relu",
"architectures": [
"FineBERTModel"
],
"batch_size": 8,
"class_name": "FineBERT",
"classifier_layers": [],
"device": "cuda",
"dropout": 0.4,
"early_stopping_patience": 8,
"freeze_bert": false,
"is_modern": false,
"learning_rate": 2e-05,
"log_dir": "logs/",
"loss": "BCELoss",
"loss_fn": "BCELoss",
"max_length": 512,
"model_name": "BiGenderDetection",
"model_save_path": "saved_models/",
"model_type": "finebert",
"num_epochs": 25,
"optimizer": "AdamW",
"output_dim": 1,
"pad_token_id": 0,
"pretrained_model": "dccuchile/bert-base-spanish-wwm-cased",
"scheduler_factor": 0.5,
"scheduler_patience": 2,
"tokenizer_name": "dccuchile/bert-base-spanish-wwm-cased",
"torch_dtype": "float32",
"transformers_version": "4.48.1",
"warmup_steps": 0,
"weight_decay": 0.01
}
|