File size: 387 Bytes
177e9e1 853b848 872e9b6 7fc9804 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
---
license: mit
---
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()
|