Upload folder using huggingface_hub

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/config.yaml

@@ -0,0 +1,115 @@
+dataset:
+  root_dir: /mnt/cvlab/scratch/cvlab/home/hantzhan/data/ts/TS_heart_cropped_resize_train
+  name: ts_latent
+  roi_x: 64
+  roi_y: 64
+  roi_z: 64
+  batch_size: 12
+  num_workers: 16
+  check_data_integrity: false
+  dist: false
+  diffusion_img_size: 64
+  diffusion_depth_size: 64
+model:
+  diffusion_num_channels: 1
+  diffusion_dim: 64
+  out_dim: 7
+  dim_mults:
+  - 1
+  - 2
+  - 4
+  - 8
+  use_bert_text_cond: false
+  use_mask_cond: false
+  use_mask_guide: false
+  results_folder: EXP/ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft
+  results_folder_postfix: ts_hybrid_diffusion_swinunetr_objv_nofft
+  load_milestone: false
+  batch_size: 12
+  num_workers: 2
+  logger: wandb
+  objective: v
+  save_and_sample_every: 10000
+  denoising_fn: hybrid_swinunetr
+  train_lr: 0.0001
+  timesteps: 100
+  loss_type: l1
+  train_num_steps: 200000
+  gradient_accumulate_every: 2
+  ema_decay: 0.995
+  amp: false
+  num_sample_rows: 4
+  gpus: 0
+  world_size: 4
+  img_channels: 1
+  mask_channels: 6
+  img_resolution: 64
+  mask_resolution: 64
+  base_resolution: 64
+  attn_heads: 8
+  attn_dim_head: 32
+  resnet_groups: 8
+  residual_scale_init: 0.1
+  img_loss_weight: 1.0
+  mask_loss_weight: 1.0
+  use_decoder_fusion: false
+  sdf_pyramid_antialias: true
+  sdf_pyramid_base_ch: 48
+  sdf_tau_vox: 20
+  use_space_to_depth_sdf: true
+  space_to_depth_out_ch: 64
+  use_edge_aware_gate: true
+  edge_gate_mode: channel
+  use_t_aware_gate: true
+  use_laplacian_pyramid: false
+  use_scale_aware_film: false
+  aux_losses:
+    sign_narrowband: false
+    multiscale_consistency: false
+    eikonal_light: false
+    tau_vox: 3
+    lambdas:
+      sign: 0.1
+      ms: 0.05
+      eik: 0.05
+  use_dmfft_img: false
+  use_dmfft_seg: false
+  lr_decay: false
+  stage_two: false
+  mm_dit:
+    model_config: MM-DiT-XL
+    fusion_type: adaptive_film
+    conditioning_type: film
+    use_flash_attn: true
+    use_rope: false
+    patch_size: 2
+  enable_test_callback: true
+  test_callback_max_samples: 2
+  test_callback_z_indices: null
+vae:
+  embed_dim: 8
+  n_embed: 512
+  pvqvae_ckpt: pretrained_models/pvqvae.pth
+  ddconfig:
+    double_z: true
+    z_channels: 8
+    resolution: 64
+    in_channels: 1
+    out_ch: 1
+    ch: 32
+    ch_mult:
+    - 1
+    - 2
+    - 4
+    - 4
+    num_res_blocks: 2
+    attn_resolutions: []
+    dropout: 0.0
+vqgan:
+  enabled: false
+  ckpt_path: pretrained_models/vqgan.ckpt
+  input_size: 64
+  latent_size: 16
+  latent_channels: 8
+  normalize_latents: true
+seed: 42

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-0.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6874af38695d9308064eb5a1617f37641e2e7d719f8572d1f15204f5e03d14e2
+size 864453250

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-1.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a2d9968e3e0130fcd368700dee1e433d5e96d55d0ec060cd40e00ccb3cd5176f
+size 864453250

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-10.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2ed882847cd8e2d95815ab1a3eed297619f42769f4efb205119a8df6f1ce1774
+size 864454686

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-11.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:464d4b234ee03da93dd2b7c2486bfd8483a3f52942dda61f8e83c5002e7ad23a
+size 864454686

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-12.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c39c2012a68889c972f57483ebddd7332a0595f6bd91ff5d1888fe1bd58cecd7
+size 864454686

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-13.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3556e984b1aacf9c67de580d61f374060eef5349c5c88f76d7f9cfc5b2df7849
+size 864454686

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-14.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d3ce2295eebfb496c99015b96503d53e50bf8ff62f97fcb4d0a477092a43d42
+size 864454686

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-15.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:45988afab906fac7e15a69e444996311e275b9074cb1d9706c63f7fc88b8f23e
+size 864454686

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-16.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5594ac2baa468b297d009451b2029536dafd1a020dd9872e0d1319688b43b2a1
+size 864454686

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-17.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c324b86b7f8a13cf6ce9a56ff6141e85c8a69e74ff582b804e7a890722eddd94
+size 864454686

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-18.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0664671241dd442ea74810ae8e86e00efe65691a3f9de07af3cd4cc468bca204
+size 864454686

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-19.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:29f29404e6d767aff85a810b6e094b9354f55193d9cfbbbf65e113534fda9b86
+size 864454686

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-2.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:923c949080ba388b10fbcd3c4ef73220c3ca2082be2af47db752f70e05c8d8cd
+size 864453250

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-3.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:62135885db027d28b21806a471a5d9e5f442b32cbcbe949c7da3696799deef12
+size 864453250

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-4.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fd353a99f55d7499fd545b63a6177424164560adabb849b9b6dbbd76880ea52b
+size 864453250

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-5.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:29c8a6d1cd863189f26e970e63380b89fdb1e463c7bf63bfc8639ecaacf5bb33
+size 864453250

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-6.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8a270d44eaf6a9c81df98f2bf38133ce5e34f24072667792faf72b704fc49ef4
+size 864453250

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-7.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e14209a67e2013f7a73f1243343957a944178da695b5b8a30c53506139dcfc10
+size 864453250

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-8.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:58316eddbf68dd07e1ce849af3fad6cf56e8cd2357c1a6bc763ae801104251d9
+size 864453250

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/model-9.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a519a7e60a96f6eb97f2163dc2f429ae2a3e45de89cb42d77d901dd0d108738e
+size 864453250

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/test_callback/milestone_0/info.json

@@ -0,0 +1,12 @@
+{
+  "milestone": 0,
+  "task_type": "unconditional",
+  "num_samples": 2,
+  "generated_shape": [
+    2,
+    2,
+    64,
+    64,
+    64
+  ]
+}

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/test_callback/milestone_1/info.json

@@ -0,0 +1,12 @@
+{
+  "milestone": 1,
+  "task_type": "unconditional",
+  "num_samples": 2,
+  "generated_shape": [
+    2,
+    2,
+    64,
+    64,
+    64
+  ]
+}

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/test_callback/milestone_10/info.json

@@ -0,0 +1,12 @@
+{
+  "milestone": 10,
+  "task_type": "unconditional",
+  "num_samples": 2,
+  "generated_shape": [
+    2,
+    2,
+    64,
+    64,
+    64
+  ]
+}

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/test_callback/milestone_11/info.json

@@ -0,0 +1,12 @@
+{
+  "milestone": 11,
+  "task_type": "unconditional",
+  "num_samples": 2,
+  "generated_shape": [
+    2,
+    2,
+    64,
+    64,
+    64
+  ]
+}

ts_latent/ts_hybrid_diffusion_swinunetr_objv_nofft/test_callback/milestone_12/info.json

@@ -0,0 +1,12 @@
+{
+  "milestone": 12,
+  "task_type": "unconditional",
+  "num_samples": 2,
+  "generated_shape": [
+    2,
+    2,
+    64,
+    64,
+    64
+  ]
+}