| { | |
| "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 | |
| } | |