prompt-safety-bert / config_used.json
Solegon's picture
Upload config_used.json with huggingface_hub
12dd853 verified
{
"model_name": "distilbert-base-uncased",
"max_length": 256,
"batch_size": 32,
"epochs": 5,
"learning_rate": 2e-05,
"weight_decay": 0.01,
"test_size": 0.15,
"random_seed": 42,
"save_dir": "saved_model",
"max_samples_per_class": 8000
}