lat_mean=39.951163 lat_std=0.000596 lon_mean=-75.19156 lon_std=0.000609

class CustomViTModel(ViTForImageClassification, PyTorchModelHubMixin): def init(self, config): super().init(config) self.classifier = nn.Linear(self.classifier.in_features, config.num_labels)

model = CustomViTModel.from_pretrained("img2gpspenn/vit") model.eval()

Downloads last month: 0

Safetensors

Model size

5.52M params

Tensor type

F32

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support