{
  "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
  }
}