Veldrovive
commited on
Commit
•
91ab8d7
1
Parent(s):
9d8e95a
Upload decoder_config.json with huggingface_hub
Browse files- decoder_config.json +5 -5
decoder_config.json
CHANGED
@@ -27,7 +27,7 @@
|
|
27 |
"dim_mults": [1, 2, 3, 4],
|
28 |
"num_resnet_blocks": 3,
|
29 |
"sparse_attn": false,
|
30 |
-
"memory_efficient":
|
31 |
"self_attn": [false, false, false, false]
|
32 |
}
|
33 |
],
|
@@ -42,7 +42,7 @@
|
|
42 |
"webdataset_base_url": "pipe:aws s3 cp --quiet s3://s-datasets/laion-aesthetic/data/laion2B-en-aesthetic/{}.tar -",
|
43 |
"img_embeddings_url": "s3://s-datasets/laion-aesthetic/ordered_embeddings/",
|
44 |
"num_workers": 6,
|
45 |
-
"batch_size":
|
46 |
"start_shard": 0,
|
47 |
"end_shard": 5247,
|
48 |
"shard_width": 5,
|
@@ -68,11 +68,11 @@
|
|
68 |
"lr": 1.2e-4,
|
69 |
"wd": 0.0,
|
70 |
"max_grad_norm": 0.5,
|
71 |
-
"save_every_n_samples":
|
72 |
"n_sample_images": 10,
|
73 |
"device": "cuda:0",
|
74 |
-
"epoch_samples":
|
75 |
-
"validation_samples":
|
76 |
"use_ema": true,
|
77 |
"ema_beta": 0.9999,
|
78 |
"save_all": false,
|
|
|
27 |
"dim_mults": [1, 2, 3, 4],
|
28 |
"num_resnet_blocks": 3,
|
29 |
"sparse_attn": false,
|
30 |
+
"memory_efficient": false,
|
31 |
"self_attn": [false, false, false, false]
|
32 |
}
|
33 |
],
|
|
|
42 |
"webdataset_base_url": "pipe:aws s3 cp --quiet s3://s-datasets/laion-aesthetic/data/laion2B-en-aesthetic/{}.tar -",
|
43 |
"img_embeddings_url": "s3://s-datasets/laion-aesthetic/ordered_embeddings/",
|
44 |
"num_workers": 6,
|
45 |
+
"batch_size": 3,
|
46 |
"start_shard": 0,
|
47 |
"end_shard": 5247,
|
48 |
"shard_width": 5,
|
|
|
68 |
"lr": 1.2e-4,
|
69 |
"wd": 0.0,
|
70 |
"max_grad_norm": 0.5,
|
71 |
+
"save_every_n_samples": 400000,
|
72 |
"n_sample_images": 10,
|
73 |
"device": "cuda:0",
|
74 |
+
"epoch_samples": 2000000,
|
75 |
+
"validation_samples": 60000,
|
76 |
"use_ema": true,
|
77 |
"ema_beta": 0.9999,
|
78 |
"save_all": false,
|