{ "base_model": "distilbert-base-uncased", "num_labels": 3, "max_length": 128, "num_train_epochs": 5, "learning_rate": 2e-05, "batch_size_train": 16, "weight_decay": 0.01, "warmup_ratio": 0.1, "fp16": true, "early_stopping_patience": 2, "metric_for_best_model": "f1", "train_size": 490, "val_size": 105, "test_size": 106 }