Upload b96s25m90/picotrust_b96s25m90.yaml with huggingface_hub

b96s25m90/picotrust_b96s25m90.yaml

@@ -0,0 +1,99 @@
+# PicoTrust b96s25m90 — 96 bits, strength 0.025, Sobel mask floor 0.90
+#
+# Independent 96-bit model for multi-model routing architecture.
+# Based on b72s20m85 production approach: Sobel texture mask, no blur.
+# Higher strength and gentler mask floor to accommodate 33% more bits.
+#
+# LDPC(96,50) = 50 payload bits (1.1 quadrillion IDs)
+
+experiment_name: picotrust_b96s25m90
+
+model:
+  type: picotrust
+  encoder_size: 512
+  decoder_size: 512
+
+data:
+  source: folder
+  path: ./data/train
+  batch_size: 4
+
+training:
+  num_steps: 160000
+  lr: 0.0001
+  num_bits: 96
+  image_size: 512
+  warmup_steps: 500
+  no_im_loss_steps: 10000
+  generator_grad_clip: 0.25
+  stn_lr_scale: 0.01
+  borders: black
+  rnd_trans: 0.10
+  rnd_trans_ramp: 10000
+  phase2_step: 60000
+  phase2_decoder_lr_scale: 0.1
+  residual_strength: 1.0
+  residual_strength_anneal_target: 0.025
+  residual_strength_anneal_start: 10000
+  residual_strength_anneal_steps: 90000
+  anneal_schedule: exponential
+  decoder_blur_sigma: 0.0
+  perceptual_blur_sigma: 0.0
+  use_sobel_mask: true
+  sobel_blur_sigma: 5.0
+  sobel_mask_floor: 0.90
+
+loss:
+  message:
+    scale: 5.0
+    ramp_steps: 1
+  l2:
+    scale: 1.5
+    ramp_steps: 50000
+  lpips:
+    scale: 1.5
+    ramp_steps: 50000
+  ffl:
+    scale: 1.0
+    ramp_steps: 50000
+    delay_steps: 0
+  message_loss_type: mse
+  l2_edge_gain: 0.0
+  use_border_falloff: true
+  gan_config:
+    enabled: true
+    discriminator_lr: 0.000015
+    g_loss_scale: 1.5
+    g_loss_ramp_steps: 20000
+
+distortion:
+  strategy: curriculum
+  perspective:
+    strength: 0.1
+    ramp_steps: 10000
+  brightness:
+    strength: 0.3
+    ramp_steps: 1000
+  saturation:
+    strength: 1.0
+    ramp_steps: 1000
+  hue:
+    strength: 0.1
+    ramp_steps: 1000
+  noise:
+    strength: 0.02
+    ramp_steps: 1000
+  jpeg_quality:
+    strength: 25
+    ramp_steps: 1000
+  enable_jpeg: true
+  enable_blur: true
+
+checkpoint:
+  dir: checkpoints
+  save_every_steps: 10000
+  keep_last: 3
+
+logging:
+  backends: [console, tensorboard]
+  log_every_steps: 100