Hugging Face model: parlange/cait-gravit-s2
Author: parlange
Created: 2025-09-06 22:07:06
Last modified: 2025-09-06 22:07:29
Pipeline: image-classification
Library: timm
Likes: 0 | Downloads: 17
Tags: timm, pytorch, safetensors, vision-transformer, image-classification, cait, gravitational-lensing, strong-lensing, astronomy, astrophysics, dataset:parlange/gravit-c21, arxiv:2509.00226, license:apache-2.0, model-index, region:us
Card metadata:
license: apache-2.0
tags: vision-transformer, image-classification, pytorch, timm, cait, gravitational-lensing, strong-lensing, astronomy, astrophysics
datasets: parlange/gravit-c21
metrics: accuracy, auc, f1
paper: title: GraViT: A Gravitational Lens Discovery Toolkit with Vision Transformers
url: https://arxiv.org/abs/2509.00226
authors: Parlange et al.