| { |
| "augmentation": { |
| "artifact_codecs": [ |
| "JPEG", |
| "WEBP" |
| ], |
| "artifact_probability": 0.15, |
| "artifact_quality_max": 98, |
| "artifact_quality_min": 88, |
| "brightness_max": 1.1, |
| "brightness_min": 0.9, |
| "brightness_probability": 0.3, |
| "contrast_max": 1.1, |
| "contrast_min": 0.9, |
| "contrast_probability": 0.3 |
| }, |
| "counts": { |
| "test_safety": { |
| "hot": 1344, |
| "nsfw": 1559, |
| "safe": 20712 |
| }, |
| "test_screen": { |
| "audio": 111, |
| "calendar": 101, |
| "camera": 41, |
| "comments": 87, |
| "contacts": 33, |
| "email": 68, |
| "excel": 40, |
| "feed": 381, |
| "finances": 840, |
| "food": 285, |
| "gallery": 66, |
| "game": 330, |
| "generic": 3820, |
| "maps": 129, |
| "message-chat": 207, |
| "message-list": 38, |
| "notes": 59, |
| "other": 15549, |
| "post": 685, |
| "profile": 174, |
| "reading": 109, |
| "shopping": 381, |
| "streaming": 51, |
| "weather": 30 |
| }, |
| "train_safety": { |
| "hot": 14997, |
| "nsfw": 18086, |
| "safe": 234685 |
| }, |
| "train_screen": { |
| "audio": 1262, |
| "calendar": 1152, |
| "camera": 477, |
| "comments": 993, |
| "contacts": 374, |
| "email": 781, |
| "excel": 455, |
| "feed": 4323, |
| "finances": 9533, |
| "food": 3238, |
| "gallery": 753, |
| "game": 3748, |
| "generic": 43298, |
| "maps": 1468, |
| "message-chat": 2346, |
| "message-list": 442, |
| "notes": 678, |
| "other": 176223, |
| "post": 7773, |
| "profile": 1975, |
| "reading": 1246, |
| "shopping": 4324, |
| "streaming": 579, |
| "weather": 344 |
| }, |
| "val_safety": { |
| "hot": 1334, |
| "nsfw": 1696, |
| "safe": 20585 |
| }, |
| "val_screen": { |
| "audio": 111, |
| "calendar": 101, |
| "camera": 41, |
| "comments": 87, |
| "contacts": 33, |
| "email": 68, |
| "excel": 40, |
| "feed": 381, |
| "finances": 840, |
| "food": 285, |
| "gallery": 66, |
| "game": 330, |
| "generic": 3820, |
| "maps": 129, |
| "message-chat": 207, |
| "message-list": 38, |
| "notes": 59, |
| "other": 15549, |
| "post": 685, |
| "profile": 174, |
| "reading": 109, |
| "shopping": 381, |
| "streaming": 51, |
| "weather": 30 |
| } |
| }, |
| "dataset": { |
| "path": "output/dataset/phone-screenshots", |
| "repo": "yapwithai/phone-screenshots", |
| "safety_field": "safety", |
| "screen_field": "label" |
| }, |
| "evaluation": { |
| "balanced_eval_max_per_class": 1000 |
| }, |
| "labels": { |
| "safety": [ |
| "hot", |
| "nsfw", |
| "safe" |
| ], |
| "screen": [ |
| "audio", |
| "calendar", |
| "camera", |
| "comments", |
| "contacts", |
| "email", |
| "excel", |
| "feed", |
| "finances", |
| "food", |
| "gallery", |
| "game", |
| "generic", |
| "maps", |
| "message-chat", |
| "message-list", |
| "notes", |
| "other", |
| "post", |
| "profile", |
| "reading", |
| "shopping", |
| "streaming", |
| "weather" |
| ] |
| }, |
| "loss": { |
| "safety": "cross_entropy", |
| "safety_batch_probability": 0.35, |
| "safety_loss_weight": 0.5, |
| "screen": "cross_entropy" |
| }, |
| "model": "timm/mobilenetv4_conv_medium.e250_r384_in12k", |
| "optimizer": { |
| "backbone_lr": 5.9999999999999995e-05, |
| "head_lr": 0.0003, |
| "lr": 0.0003, |
| "weight_decay": 0.05 |
| }, |
| "preprocess": { |
| "crop": false, |
| "horizontal_flip": false, |
| "mean": [ |
| 0.485, |
| 0.456, |
| 0.406 |
| ], |
| "preserve_aspect_ratio": true, |
| "resize_longest_side_px": 1024, |
| "std": [ |
| 0.229, |
| 0.224, |
| 0.225 |
| ], |
| "stretch": false |
| }, |
| "sampling": { |
| "folded_screen_labels": {}, |
| "min_train_count": 300, |
| "mode": "adaptive_label_aspect_bucketed", |
| "safety_max_repeat": 8, |
| "safety_target_ratio": 2.0, |
| "screen_max_repeat": 6, |
| "screen_target_ratio": 5.0 |
| }, |
| "schema": 3, |
| "training": { |
| "batch_size": 8, |
| "effective_batch_size": 64, |
| "epochs": 4, |
| "grad_accum_steps": 8, |
| "seed": 1337, |
| "target_effective_batch_size": 64, |
| "workers": 4 |
| } |
| } |
|
|
