Veldrovive commited on
Commit
91ab8d7
1 Parent(s): 9d8e95a

Upload decoder_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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": true,
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": 7,
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": 1000000,
72
  "n_sample_images": 10,
73
  "device": "cuda:0",
74
- "epoch_samples": 5500000,
75
- "validation_samples": 100000,
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,