Add VAE checkpoint
Browse files- vae/vae_2_1/config.json +21 -0
- vae/vae_2_1/pytorch_model.ckpt +3 -0
    	
        vae/vae_2_1/config.json
    ADDED
    
    | @@ -0,0 +1,21 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "_class_name": "HYVAE2D",
         | 
| 3 | 
            +
              "block_out_channels": [
         | 
| 4 | 
            +
                128,
         | 
| 5 | 
            +
                256,
         | 
| 6 | 
            +
                512,
         | 
| 7 | 
            +
                512,
         | 
| 8 | 
            +
                1024,
         | 
| 9 | 
            +
                1024
         | 
| 10 | 
            +
              ],
         | 
| 11 | 
            +
              "in_channels": 3,
         | 
| 12 | 
            +
              "out_channels": 3,
         | 
| 13 | 
            +
              "latent_channels": 64,
         | 
| 14 | 
            +
              "layers_per_block": 2,
         | 
| 15 | 
            +
              "ffactor_spatial": 32,
         | 
| 16 | 
            +
              "sample_size": 384,
         | 
| 17 | 
            +
              "sample_tsize": 96,
         | 
| 18 | 
            +
              "scaling_factor": 0.75289,
         | 
| 19 | 
            +
              "downsample_match_channel": true,
         | 
| 20 | 
            +
              "upsample_match_channel": true
         | 
| 21 | 
            +
            }
         | 
    	
        vae/vae_2_1/pytorch_model.ckpt
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:85a0aec2ffb92ba1bba363c6e9c673e1b52997808a4a496e2b127b1e62733e90
         | 
| 3 | 
            +
            size 1622406234
         | 
