Upload folder using huggingface_hub
#1
by TimMerino1710 - opened
- config.json +73 -0
- inference_weights/model_best_weights.pt +3 -0
config.json
ADDED
|
@@ -0,0 +1,73 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"adam_betas": "0.9,0.99",
|
| 3 |
+
"amp": true,
|
| 4 |
+
"attn_drop": 0.0,
|
| 5 |
+
"auto_normalize": false,
|
| 6 |
+
"beta_schedule": "cosine",
|
| 7 |
+
"comb_method": "add",
|
| 8 |
+
"compile_model": false,
|
| 9 |
+
"cond_drop_prob": 0.2,
|
| 10 |
+
"config": null,
|
| 11 |
+
"crop": true,
|
| 12 |
+
"crop_size": 32,
|
| 13 |
+
"data_channels": 16,
|
| 14 |
+
"dataloader_num_workers": 4,
|
| 15 |
+
"dataloader_persistent_workers": true,
|
| 16 |
+
"dataloader_prefetch_factor": 2,
|
| 17 |
+
"dataset_path": "data/balanced_dataset/32_biome_parts_dir_balanced_cc_dir",
|
| 18 |
+
"ddim_sampling_eta": 0.0,
|
| 19 |
+
"default_cond_scale": 4.0,
|
| 20 |
+
"depth": 25,
|
| 21 |
+
"device": "cuda",
|
| 22 |
+
"ema_decay": 0.995,
|
| 23 |
+
"ema_update_every": 10,
|
| 24 |
+
"embeddings_dict_key": null,
|
| 25 |
+
"embeddings_path": "assets/full_embeddings_16_raw.pt",
|
| 26 |
+
"evaluate_ema_model": false,
|
| 27 |
+
"exp_name": "DDPM_1e4_balanced",
|
| 28 |
+
"gradient_accumulate_every": 4,
|
| 29 |
+
"hidden_channels": 768,
|
| 30 |
+
"image_size": 32,
|
| 31 |
+
"mappings_file_path": "data/balanced_dataset/32_biome_parts_dir_balanced_cc_oh_mappings.pt",
|
| 32 |
+
"max_grad_norm": 1.0,
|
| 33 |
+
"min_lr_ratio": 0.0,
|
| 34 |
+
"min_snr_gamma": 5.0,
|
| 35 |
+
"min_snr_loss_weight": true,
|
| 36 |
+
"mixed_precision_type": "bf16",
|
| 37 |
+
"mlp_ratio": 4.0,
|
| 38 |
+
"num_classes": 15,
|
| 39 |
+
"num_heads": 8,
|
| 40 |
+
"num_samples": 15,
|
| 41 |
+
"objective": "pred_v",
|
| 42 |
+
"offset_noise_strength": 0.0,
|
| 43 |
+
"optimizer": "Adam",
|
| 44 |
+
"patch_size": 2,
|
| 45 |
+
"per_device_train_batch_size": 32,
|
| 46 |
+
"proj_drop": 0.0,
|
| 47 |
+
"results_folder": "./results",
|
| 48 |
+
"rotation_aug_prob": 0.75,
|
| 49 |
+
"run_name": "DDPM_crop_32_full_balanced_dataset",
|
| 50 |
+
"sampling_timesteps": 1000,
|
| 51 |
+
"save_and_sample_every": 3514,
|
| 52 |
+
"save_and_sample_every_epoch": 2,
|
| 53 |
+
"save_only_last_checkpoint": true,
|
| 54 |
+
"scheduler": "cosine",
|
| 55 |
+
"split_batches": false,
|
| 56 |
+
"time_dim": 256,
|
| 57 |
+
"timesteps": 1000,
|
| 58 |
+
"train_batch_size": 512,
|
| 59 |
+
"train_lr": 0.001,
|
| 60 |
+
"train_num_epochs": 20,
|
| 61 |
+
"train_num_steps": 35140,
|
| 62 |
+
"use_wandb": true,
|
| 63 |
+
"val_dataset_path": "data/balanced_dataset/32_biome_parts_dir_balanced_cc_val_dir",
|
| 64 |
+
"val_mappings_file_path": "data/balanced_dataset/32_biome_parts_dir_balanced_cc_oh_mappings.pt",
|
| 65 |
+
"validate_every": 3514,
|
| 66 |
+
"validate_every_epoch": 2,
|
| 67 |
+
"wandb_group_name": "DDPM",
|
| 68 |
+
"wandb_log_images": true,
|
| 69 |
+
"wandb_project": "dream-cubed",
|
| 70 |
+
"wandb_run_name": "DDPM_crop_32_full_balanced_dataset",
|
| 71 |
+
"warmup_steps": 0,
|
| 72 |
+
"weight_decay": 0.01
|
| 73 |
+
}
|
inference_weights/model_best_weights.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d843e30f778b9eb0ff08cf8d51bec19d4f736d136f79e841e5fb689a007d26da
|
| 3 |
+
size 1082527083
|