update content with the text model from Thomas repository https://huggingface.co/spaces/tombou/frugal-ai-challenge
42b7ac6
| { | |
| "mode": "train", | |
| "model_type": "distilbert", | |
| "model_name": "distilbert-base-uncased", | |
| "batch_size": 16, | |
| "num_epochs": 5, | |
| "initial_learning_rate": 2e-5 | |
| } | |