latterworks commited on
Commit
690fda8
·
verified ·
1 Parent(s): b0a0435

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +21 -0
config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tile_size": 256,
3
+ "latent_channels": 8,
4
+ "latent_size": 32,
5
+ "satclip_dim": 256,
6
+ "prior_hidden_dim": 256,
7
+ "refine_base_channels": 96,
8
+ "num_timesteps": 500,
9
+ "beta_start": 0.0001,
10
+ "beta_end": 0.02,
11
+ "batch_size": 16,
12
+ "lr": 0.0003,
13
+ "num_epochs": 5,
14
+ "vae_epochs": 30,
15
+ "vae_lr": 0.0001,
16
+ "vae_kl_weight": 1e-05,
17
+ "prior_loss_weight": 1.0,
18
+ "diffusion_loss_weight": 1.0,
19
+ "device": "cuda",
20
+ "num_workers": 8
21
+ }