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
- 41