pereza commited on
Commit
0ec2906
1 Parent(s): 6964cb6
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ samples/VQModel_0005.png filter=lfs diff=lfs merge=lfs -text
37
+ samples/VQModel_0010.png filter=lfs diff=lfs merge=lfs -text
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "VQModel",
3
+ "_diffusers_version": "0.17.1",
4
+ "act_fn": "silu",
5
+ "block_out_channels": [
6
+ 8,
7
+ 16,
8
+ 16
9
+ ],
10
+ "down_block_types": [
11
+ "DownEncoderBlock2D",
12
+ "AttnDownEncoderBlock2D",
13
+ "AttnDownEncoderBlock2D"
14
+ ],
15
+ "in_channels": 1,
16
+ "latent_channels": 1,
17
+ "layers_per_block": 1,
18
+ "norm_num_groups": 8,
19
+ "norm_type": "group",
20
+ "num_vq_embeddings": 256,
21
+ "out_channels": 1,
22
+ "sample_size": [
23
+ 160,
24
+ 240
25
+ ],
26
+ "scaling_factor": 0.18215,
27
+ "up_block_types": [
28
+ "UpDecoderBlock2D",
29
+ "AttnUpDecoderBlock2D",
30
+ "AttnUpDecoderBlock2D"
31
+ ],
32
+ "vq_embed_dim": 32
33
+ }
diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88a7cc3cae8912d5ceb4578585edc19e56a8b86ec212b14d1405979947d52c54
3
+ size 380295
logs/Train VQ-VAE NN/1688547358.5822568/events.out.tfevents.1688547358.planet-express.3223273.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3710f2be1696668d5322e943b8c9bc3fc987ddf5e7a57a4485f6cd1961ee1566
3
+ size 1044
logs/Train VQ-VAE NN/1688547358.5914624/hparams.yml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ __pydantic_initialised__: true
2
+ device: cuda
3
+ hour_embed_type: class
4
+ hub_private_repo: false
5
+ learning_rate: 0.0001
6
+ lr_warmup_steps: 500
7
+ mixed_precision: fp16
8
+ num_epochs: 200
9
+ num_samples: 3
10
+ output_dir: vqvae-tiny-small
11
+ overwrite_output_dir: true
12
+ push_to_hub: true
13
+ save_image_epochs: 5
14
+ save_model_epochs: 10
15
+ seed: 2023
logs/Train VQ-VAE NN/1688551645.242862/events.out.tfevents.1688551645.planet-express.3244353.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7d113ca037b10483f8913a1c02af6df2b97ff84f3caa01d32a5ca9a0a035248
3
+ size 1044
logs/Train VQ-VAE NN/1688551645.2525907/hparams.yml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ __pydantic_initialised__: true
2
+ device: cuda
3
+ hour_embed_type: class
4
+ hub_private_repo: false
5
+ learning_rate: 0.0001
6
+ lr_warmup_steps: 500
7
+ mixed_precision: fp16
8
+ num_epochs: 200
9
+ num_samples: 3
10
+ output_dir: vqvae-tiny-small
11
+ overwrite_output_dir: true
12
+ push_to_hub: true
13
+ save_image_epochs: 5
14
+ save_model_epochs: 10
15
+ seed: 2023
logs/Train VQ-VAE NN/1688566467.6950643/events.out.tfevents.1688566467.planet-express.3302752.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d2b13ad3d30a775d3b5229f9323e88807251276b1e149df88f4a7750e732a21
3
+ size 1044
logs/Train VQ-VAE NN/1688566467.7050855/hparams.yml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ __pydantic_initialised__: true
2
+ device: cuda
3
+ hour_embed_type: class
4
+ hub_private_repo: false
5
+ learning_rate: 0.0001
6
+ lr_warmup_steps: 500
7
+ mixed_precision: fp16
8
+ num_epochs: 200
9
+ num_samples: 3
10
+ output_dir: vqvae-tiny-small
11
+ overwrite_output_dir: true
12
+ push_to_hub: true
13
+ save_image_epochs: 5
14
+ save_model_epochs: 10
15
+ seed: 2023
logs/Train VQ-VAE NN/events.out.tfevents.1688547358.planet-express.3223273.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4eb9d04d0a268141baf273a0f99f452833baadc3cc21a355d6bf9fde59b330e7
3
+ size 88
logs/Train VQ-VAE NN/events.out.tfevents.1688551645.planet-express.3244353.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14a1e4f7083d8a810df820b83c563672edec144a73d203b4087a867d58a67bbe
3
+ size 299818000
logs/Train VQ-VAE NN/events.out.tfevents.1688566467.planet-express.3302752.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74c8e753d19422653244cdf62fb17a19959bddbb6c496e1cb6607f1affb55ded
3
+ size 3025874822
samples/VQModel_0005.png ADDED

Git LFS Details

  • SHA256: 48c8a5894a8df79b13472c17c708c9f995913e73876d9c327fb5e078a9d52979
  • Pointer size: 130 Bytes
  • Size of remote file: 81.5 kB
samples/VQModel_0010.png ADDED

Git LFS Details

  • SHA256: e1a5a0cd98e0ad0084c66c79c6929372fb03110b607c33968a2fbe63cb5d68fb
  • Pointer size: 130 Bytes
  • Size of remote file: 82 kB