Update conv_unet.yaml

lhallee · lhallee · commit 652e75fb3db0 · 2026-02-06T10:24:30.000-05:00
diff --git a/example_yamls/conv_unet.yaml b/example_yamls/conv_unet.yaml
@@ -43,9 +43,9 @@ dynamo_recompile_limit: 32
 mlm: true
 masked_diffusion: false
 mask_rate: 0.2
-starting_mask_rate: 0.1
+starting_mask_rate: 0.2
 mask_rate_steps: 2500
-mask_rate_schedule: true
+mask_rate_schedule: false
 
 # Optimization & Schedule
 batch_size: 1048576     # Total tokens across all GPUs
