| { |
| "model_type": "divae", |
| "image_size": 448, |
| "patch_size": 16, |
| "n_channels": 2, |
| "enc_type": "vit_b_enc", |
| "dec_type": "unet_patched", |
| "quant_type": "fsq", |
| "codebook_size": "8-8-8-6-5", |
| "latent_dim": 5, |
| "commitment_weight": 1.0, |
| "post_mlp": true, |
| "norm_codes": true, |
| "num_train_timesteps": 1000, |
| "beta_schedule": "linear", |
| "prediction_type": "sample", |
| "zero_terminal_snr": true, |
| "modality": "ahn", |
| "normalization": { |
| "kind": "minmax", |
| "clip_min": -20.0, |
| "clip_max": 80.0 |
| }, |
| "geospatial_window_px": 448, |
| "token_grid_size": 28, |
| "vocab_size": 15360, |
| "align_with_pixel_size_m": 0.08 |
| } |
|
|
