fine tuned nsfw classification This model is a fine tuned version of Falconsai/nsfw image detection on the imagefolder dataset. It achieves the following results on the evaluation set: Loss: 0.0000 Accuracy: 1.0 Model description More information needed Intended uses & limitations More information needed Training and evaluation data More information needed Training procedure Training hyperparameters The following hyperparameters were used during training: learning rate: 5e 05 train 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: 4 Training results Training Loss Epoch Step Validation Loss Accuracy : : : : : : : : : : 0.0176 0.6083 500 0.0093 0.999 0.0031 1.2165 1000 0.0021 0.999 0.0015 1.8248 1500 0.0049 0.999 0.0001 2.4331 2000 0.0086 0.999 0.0 3.0414 2500 0.0087 0.999 0.0 3.6496 3000 0.0088 0.999 Framework versions Transformers 4.48.3 Pytorch 2.5.1+cu124 Datasets 3.3.2 Tokenizers 0.21.0
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy