FredZhang7 commited on
Commit
1e5dd4d
·
1 Parent(s): dae8b5c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -7,7 +7,7 @@ tags:
7
  # Google Safesearch Mini Model Card
8
  This model is trained on 2,224,000 images scraped from Google Images, Reddit, Imgur, and Github.
9
  It predicts the likelihood of an image being nsfw_gore, nsfw_suggestive, and safe.
10
- After 20 epochs on PyTorch's InceptionV3, the model achieves 94% acc on both train and test data. After 3.3 epochs on Keras's Xception, the model scores 94% acc on train set and 92% on test set.
11
 
12
  Using this instead of the stable diffusion safety checker allows users to save 1.12GB of RAM and disk space.
13
 
 
7
  # Google Safesearch Mini Model Card
8
  This model is trained on 2,224,000 images scraped from Google Images, Reddit, Imgur, and Github.
9
  It predicts the likelihood of an image being nsfw_gore, nsfw_suggestive, and safe.
10
+ After 20 epochs on PyTorch, the finetuned InceptionV3 model achieves 94% acc on both train and test data. After 3.3 epochs on Keras, the finetuned Xception model scores 94% acc on train set and 92% on test set.
11
 
12
  Using this instead of the stable diffusion safety checker allows users to save 1.12GB of RAM and disk space.
13