albert-sexism-classifier / training_params.json
MatteoWood's picture
Upload folder using huggingface_hub
06f4a4a
raw
history blame contribute delete
729 Bytes
{"data_path": "MatteoWood/autotrain-data-albert-sexism-classifier", "model": "albert-base-v2", "lr": 5e-05, "epochs": 5, "max_seq_length": 128, "batch_size": 8, "warmup_ratio": 0.1, "gradient_accumulation": 1, "optimizer": "adamw_torch", "scheduler": "linear", "weight_decay": 0.0, "max_grad_norm": 1.0, "seed": 42, "train_split": "train", "valid_split": "validation", "text_column": "autotrain_text", "target_column": "autotrain_label", "logging_steps": -1, "project_name": "/tmp/model", "auto_find_batch_size": false, "fp16": false, "save_total_limit": 1, "save_strategy": "epoch", "push_to_hub": true, "repo_id": "MatteoWood/albert-sexism-classifier-0", "evaluation_strategy": "epoch", "username": "MatteoWood", "log": "none"}