config.json

{
  "tile_size": 256,
  "latent_channels": 8,
  "latent_size": 32,
  "satclip_dim": 256,
  "prior_hidden_dim": 256,
  "refine_base_channels": 96,
  "num_timesteps": 500,
  "beta_start": 0.0001,
  "beta_end": 0.02,
  "batch_size": 16,
  "lr": 0.0003,
  "num_epochs": 5,
  "vae_epochs": 30,
  "vae_lr": 0.0001,
  "vae_kl_weight": 1e-05,
  "prior_loss_weight": 1.0,
  "diffusion_loss_weight": 1.0,
  "device": "cuda",
  "num_workers": 8
}