submission / config_training.json
pierre-loic's picture
update content with the text model from Thomas repository https://huggingface.co/spaces/tombou/frugal-ai-challenge
42b7ac6
raw
history blame contribute delete
167 Bytes
{
"mode": "train",
"model_type": "distilbert",
"model_name": "distilbert-base-uncased",
"batch_size": 16,
"num_epochs": 5,
"initial_learning_rate": 2e-5
}