Uploading AE in onlytojay/conv_ae_mnist

README.md

@@ -0,0 +1,13 @@
+---
+language: en
+tags:
+- pythae
+license: apache-2.0
+---
+
+### Downloading this model from the Hub
+This model was trained with pythae. It can be downloaded or reloaded using the method `load_from_hf_hub`
+```python
+>>> from pythae.models import AutoModel
+>>> model = AutoModel.load_from_hf_hub(hf_hub_path="your_hf_username/repo_name")
+```

decoder.pkl

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

encoder.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:661be051c321a6d2dcb39d5eee192f263e32debc0fc4d203c1308e8c23636459
+size 44172873

environment.json

@@ -0,0 +1 @@
+{"name": "EnvironmentConfig", "python_version": "3.11"}

model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:786d6a7d91948cb44284bd343fdfb53508752413aa5b1adf0c29e77220af6de6
+size 68898407

model_config.json

@@ -0,0 +1 @@
+{"name": "AEConfig", "input_dim": [1, 28, 28], "latent_dim": 16, "uses_default_encoder": false, "uses_default_decoder": false}