Allayte commited on
Commit
c9f6db8
1 Parent(s): 7c2798d

Uploading BetaVAE in Allayte/BETA-VAE-4-typography-10

Browse files
Files changed (6) hide show
  1. README.md +13 -0
  2. decoder.pkl +3 -0
  3. encoder.pkl +3 -0
  4. environment.json +1 -0
  5. model.pt +3 -0
  6. model_config.json +1 -0
README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - pythae
5
+ license: apache-2.0
6
+ ---
7
+
8
+ ### Downloading this model from the Hub
9
+ This model was trained with pythae. It can be downloaded or reloaded using the method `load_from_hf_hub`
10
+ ```python
11
+ >>> from pythae.models import AutoModel
12
+ >>> model = AutoModel.load_from_hf_hub(hf_hub_path="your_hf_username/repo_name")
13
+ ```
decoder.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe2c9e88be7c5a11ecb9116f789c432d04b8f38511e8abff64793511ffdabaac
3
+ size 1369948
encoder.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e5c2477333b9e9cbe5e668c7715c38598b7b7597830229569d82ab462586ae1
3
+ size 1725188
environment.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"name": "EnvironmentConfig", "python_version": "3.8"}
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:110c6cd46372075a744efe462d37ea6ad8fcfca042d18d1017863b7827f30d9a
3
+ size 3077047
model_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"name": "BetaVAEConfig", "input_dim": [1, 28, 28], "latent_dim": 16, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "mse", "beta": 2.0}