Hello,
Looking at the definition of the ViT-H-14 variant, the model does not have a sigmoid activation at the output. Are we supposed to add this, or just clamp the value between 0 and 1? It's currently unconstrained.
https://github.com/LAION-AI/CLIP-based-NSFW-Detector/blob/main/h14_nsfw_model.py#L24
Commit is from @nousr, so perhaps you might now?
Hello, Looking at the definition of the ViT-H-14 variant, the model does not have a sigmoid activation at the output. Are we supposed to add this, or just clamp the value between 0 and 1? It's currently unconstrained. https://github.com/LAION-AI/CLIP-based-NSFW-Detector/blob/main/h14_nsfw_model.py#L24 Commit is from @nousr, so perhaps you might now?