Epoch 10
Browse files- .gitattributes +2 -0
- config.json +33 -0
- diffusion_pytorch_model.bin +3 -0
- logs/Train VQ-VAE NN/1688547358.5822568/events.out.tfevents.1688547358.planet-express.3223273.1 +3 -0
- logs/Train VQ-VAE NN/1688547358.5914624/hparams.yml +15 -0
- logs/Train VQ-VAE NN/1688551645.242862/events.out.tfevents.1688551645.planet-express.3244353.1 +3 -0
- logs/Train VQ-VAE NN/1688551645.2525907/hparams.yml +15 -0
- logs/Train VQ-VAE NN/1688566467.6950643/events.out.tfevents.1688566467.planet-express.3302752.1 +3 -0
- logs/Train VQ-VAE NN/1688566467.7050855/hparams.yml +15 -0
- logs/Train VQ-VAE NN/events.out.tfevents.1688547358.planet-express.3223273.0 +3 -0
- logs/Train VQ-VAE NN/events.out.tfevents.1688551645.planet-express.3244353.0 +3 -0
- logs/Train VQ-VAE NN/events.out.tfevents.1688566467.planet-express.3302752.0 +3 -0
- samples/VQModel_0005.png +3 -0
- samples/VQModel_0010.png +3 -0
.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
|
samples/VQModel_0010.png
ADDED
Git LFS Details
|