trained_nsfw_detectionv1

README.md

@@ -14,9 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # results
 
-This model is a fine-tuned version of [aretolabs/nsfw_detection](https://huggingface.co/aretolabs/nsfw_detection) on an unknown dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.0662
+This model is a fine-tuned version of [aretolabs/nsfw_detection](https://huggingface.co/aretolabs/nsfw_detection) on the None dataset.
 
 ## Model description
 
@@ -35,28 +33,21 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 3e-05
+- learning_rate: 2e-05
 - train_batch_size: 16
-- eval_batch_size: 16
+- eval_batch_size: 8
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 5
-- mixed_precision_training: Native AMP
+- num_epochs: 3
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss |
-|:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 8    | 0.1408          |
-| No log        | 2.0   | 16   | 0.0801          |
-| No log        | 3.0   | 24   | 0.0726          |
-| No log        | 4.0   | 32   | 0.0757          |
-| No log        | 5.0   | 40   | 0.0662          |
 
 
 ### Framework versions
 
 - Transformers 4.47.1
-- Pytorch 2.5.1+cu121
+- Pytorch 2.5.1+cu124
+- Datasets 3.2.0
 - Tokenizers 0.21.0

config.json

@@ -9,13 +9,17 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "LABEL_0"
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2"
   },
   "image_size": 384,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "LABEL_0": 0
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_2": 2
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -23,7 +27,7 @@
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,
-  "problem_type": "regression",
+  "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
   "transformers_version": "4.47.1"

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d19da0acc6cd86b01c7a2d00fb935ff8d59336dfebb52763f2c9529b32c51260
-size 344388252
+oid sha256:960b617da74dd7f093c8b503717d345844ba7de92fcbab94928b95da2065c9ed
+size 344394412

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b4a2e327d743ed2daedf3691fd37a29845c15200f3fd98837a79df90e3393144
+oid sha256:6aa26332a4fe8ce0512ab8591acc75521007f9577a7da8b3ded984e29e5a7c24
 size 5304