Initial release — iter_10000 checkpoint + ag_config

ag_config.yaml

@@ -0,0 +1,52 @@
+# DDP config template (default, use with predict2_anomaly_gen_ddp_{2b,14b}).
+# Placeholders in <angle-brackets> are filled by scripts/generate_config.py.
+# Keep sections minimal — the experiment provides defaults for everything else.
+job:
+  project: anomaly_gen
+  group: UC2_exp
+  name: UC2_exp_training_FP32_lr0.02_bs=2_2B_512x512
+optimizer:
+  lr: 0.02
+checkpoint:
+  save_iter: 2000
+trainer:
+  max_iter: 75000
+  logging_iter: 10
+  validation_iter: 2000
+  run_validation: True
+  early_stop:
+    enabled: true
+    metric: nn
+    patience: 5
+    min_delta: 0
+    min_delta_mode: rel
+dataloader_train:
+  batch_size: 2
+  dataset:
+    dataset_dir: datasets/UC2_data
+    image_size:
+      - 512
+      - 512
+    anomaly_types: [[metal_surface, MT_Blowhole], [metal_surface, MT_Break], [metal_surface, MT_Crack],
+  [metal_surface, MT_Fray], [metal_surface, MT_Uneven]]
+    seed: 1
+    data_augprob: 0.5
+    aug_type: random_ratio_crop
+    ratio_range: [1.5, 8.0]
+dataloader_val:
+  batch_size: 32
+  dataset:
+    input_data_path: ag_inference/validation_UC2_exp/testcase.jsonl
+model:
+  config:
+    ag_config:
+      ad_precision: float32
+      t5_model_name: checkpoints/google-t5/t5-large
+      anomaly_embedding:
+        anomaly_types: [[metal_surface, MT_Blowhole], [metal_surface, MT_Break], [metal_surface, MT_Crack],
+  [metal_surface, MT_Fray], [metal_surface, MT_Uneven]]
+        freeze: False
+      mask_encoder:
+        encoder_config:
+          init_cfg:
+            checkpoint: checkpoints/NVDINOV2/nv_dinov2_classification_model.ckpt

iter_000010000.pt

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