config.json

{
  "architectures": [
    "SafeSearchModel"
  ],
  "auto_map": {
    "AutoConfig": "model.SafeSearchConfig",
    "AutoModelForImageClassification": "model.SafeSearchModel"
  },
  "classes": {
    "0": "nsfw_gore",
    "1": "nsfw_suggestive",
    "2": "safe"
  },
  "classifier": "default",
  "crop_pct": 0.875,
  "device": "cpu",
  "first_conv": "conv2d",
  "has_aux": false,
  "input_channels": 3,
  "input_size": [
    3,
    299,
    299
  ],
  "interpolation": "bicubic",
  "label_offset": 0,
  "mean": [
    0.5,
    0.5,
    0.5
  ],
  "model_name": "safesearch_mini_v2",
  "model_type": "safesearch_mini_v2",
  "num_classes": 3,
  "output_channels": 1536,
  "pool_size": [
    8,
    8
  ],
  "std": [
    0.5,
    0.5,
    0.5
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.25.1"
}