{ "data_path": "haiefff/nsfw-images", "model": "google/vit-base-patch16-224", "username": "haiefff", "lr": 5e-05, "epochs": 5, "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": null, "logging_steps": -1, "project_name": "nsfw-images", "auto_find_batch_size": false, "mixed_precision": "fp16", "save_total_limit": 1, "push_to_hub": true, "eval_strategy": "epoch", "image_column": "image", "target_column": "label", "log": "tensorboard", "early_stopping_patience": 5, "early_stopping_threshold": 0.01 }