Upload 2 files

depthdif_v1.ckpt

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

model_config.yaml

@@ -0,0 +1,40 @@
+# Settings for instantiating the pixel-space diffusion architecture/checkpoint behavior.
+model:
+  model_type: "cond_px_dif" # Pixel-space conditional diffusion.
+  resume_checkpoint: false # false/null starts from scratch; set to a .ckpt path to continue training.
+  load_checkpoint: "logs/selection/GLORYS_pretrained/last.ckpt" # false/null disables warm start; set to a .ckpt path to load model weights only.
+  generated_channels: 50 # Predict the full 50-level GLORYS temperature stack.
+  condition_channels: 52 # OSTIA (1) + corrupted Argo stack (50) + collapsed x_valid_mask (1).
+  condition_mask_channels: 1 # Collapse x_valid_mask across depth to keep conditioning lighter than a 50-channel mask.
+  condition_include_eo: true # Include batch["eo"] as an extra condition channel.
+  condition_use_valid_mask: true # Concatenate x_valid_mask to the condition tensor.
+  clamp_known_pixels: false # Sampling-time clamping is optional; leave it off for the default training preset.
+  mask_loss_with_valid_pixels: true # Uses task-valid supervision masks: y_valid_mask in standard mode, x_valid_mask ∩ y_valid_mask in ambient mode.
+  parameterization: "x0" # Keeps the base preset compatible with the ambient objective.
+  log_intermediates: false # Training config can still override validation logging behavior.
+  ambient_occlusion:
+    enabled: false # Base model preset keeps standard y-reconstruction over y_valid_mask.
+    further_drop_prob: 0.1 # Mild extra corruption over already-observed Argo pixels.
+    apply_to_noisy_branch: true # Applies the further-corrupted mask to the noisy target branch.
+    shared_spatial_mask: true # Shares one spatial further-mask across all depth channels.
+    min_kept_observed_pixels: 50 # Keep a non-trivial number of observed channel-pixels after further corruption.
+    require_x0_parameterization: true # Ambient objective stays tied to x0 parameterization.
+  post_process:
+    gaussian_blur:
+      enabled: true # Apply final denormalized smoothing for global export inference.
+      sigma: 0.75 # Medium blur strength selected from DDPM/DDIM visual comparison.
+      kernel_size: 5 # Even values are auto-adjusted to the next odd number.
+
+  coord_conditioning:
+    enabled: true # Enable patch-center coordinate conditioning via FiLM.
+    encoding: "unit_sphere" # Stable latitude/longitude encoding for global patches.
+    include_date: true # Also encode sample date together with coordinates.
+    date_encoding: "day_of_year_sincos" # Non-leap day-of-year sin/cos encoding.
+    embed_dim: null # Defaults to unet.dim when left null.
+
+  unet:
+    dim: 64 # Wider base width to better match the 50-output / 52-condition channel setup. Other Common values: 48 for 16x16, 64 for 32x32, 96 for 48x48, 128 for 64x64.
+    dim_mults: [1, 2, 4, 8] # Keep the established U-Net depth while reducing width.
+    with_time_emb: true # Enable timestep embeddings in the denoiser.
+    output_mean_scale: false # Optional output mean correction remains disabled.
+    residual: false # Predict the denoised output directly instead of a residual.