Push model using huggingface_hub.

README.md

@@ -0,0 +1,10 @@
+---
+pipeline_tag: image-classification
+tags:
+- model_hub_mixin
+- pytorch_model_hub_mixin
+---
+
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Library: https://github.com/ananthu-aniraj/pdiscoformer
+- Docs: [More Information Needed]

config.json

@@ -0,0 +1,13 @@
+{
+  "classifier_type": "linear",
+  "gumbel_softmax": true,
+  "gumbel_softmax_hard": false,
+  "gumbel_softmax_temperature": 1.0,
+  "modulation_orth": true,
+  "modulation_type": "layer_norm",
+  "noise_variance": 0.0,
+  "num_classes": 102,
+  "num_landmarks": 8,
+  "part_dropout": 0.3,
+  "return_transformer_qkv": false
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18b73fafe6cdc52ab47217a46619921c541f87c533186585a4da652662cccb92
+size 343321688