FredZhang7's picture
Update config.json
2fea4ae
raw
history blame
866 Bytes
{
"architectures": [
"InceptionV3ModelForImageClassification"
],
"auto_map": {
"AutoConfig": "Config.InceptionV3Config",
"AutoModelForImageClassification": "Model.InceptionV3ModelForImageClassification"
},
"classes": {
"0": "nsfw_gore",
"1": "nsfw_suggestive",
"2": "safe"
},
"classifier": "fc",
"crop_pct": 0.875,
"first_conv": "Conv2d_1a_3x3.conv",
"has_aux": true,
"input_channels": 3,
"input_size": [
3,
299,
299
],
"interpolation": "bicubic",
"label_offset": 1,
"mean": [
0.5,
0.5,
0.5
],
"model_name": "google-safesearch-mini",
"model_type": "inceptionv3",
"num_classes": 3,
"output_channels": 2048,
"pool_size": [
8,
8,
2048
],
"std": [
0.5,
0.5,
0.5
],
"torch_dtype": "float32",
"transformers_version": "4.25.1",
"use_jit": true
}