| { | |
| "model_name": "indolem/indobert-base-uncased", | |
| "num_epochs": 3, | |
| "batch_size": 64, | |
| "eval_batch_size": 128, | |
| "max_length": 128, | |
| "train_samples": 1799998, | |
| "test_samples": 200000, | |
| "precision": "bf16", | |
| "gpu": "NVIDIA A100-SXM4-40GB", | |
| "label_map": { | |
| "Islam": 0, | |
| "Catholic": 1, | |
| "Protestant": 2 | |
| } | |
| } |