Skip to content

Commit 17c8b27

Browse files
CEQ151CEQ151
CEQ151
authored and
CEQ151
committed
chore(mlwm): archive alpha1 training checkpoint
1 parent 2106e1c commit 17c8b27

4 files changed

Lines changed: 237 additions & 0 deletions

File tree

‎artifacts/promoted/mlwm-v1-alpha1/best.ckpt‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
version https://git-lfs.github.com/spec/v1
2+
oid sha256:d752bcc4708e7b74e661bafe2b95e4ac4f6d2e04da6edca918f49ac56792d9b5
3+
size 8086229

‎artifacts/promoted/mlwm-v1-alpha1/eval_full_attack.yaml‎

Lines changed: 102 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,102 @@
1+
seed: 20260426
2+
image_size: 448
3+
batch_size: 8
4+
grad_accum: 2
5+
num_workers: 12
6+
device: cuda
7+
amp: true
8+
max_text_bytes: 16
9+
payload_bits: 256
10+
train_dir: data/train_images
11+
val_dir: data/val_images
12+
artifacts_dir: artifacts/mlwm_v1
13+
promoted_dir: artifacts/promoted
14+
models_dir: resources/models/neural_wm
15+
optimizer:
16+
lr: 4.0e-05
17+
weight_decay: 1.0e-05
18+
loss:
19+
payload_weight: 10.0
20+
image_weight: 0.6
21+
tv_weight: 0.01
22+
residual_weight: 0.02
23+
stages:
24+
warmup:
25+
enabled: false
26+
epochs: 3
27+
freeze_encoder: true
28+
fixed_residual_scale: 0.04
29+
attack_strength: medium
30+
main:
31+
enabled: false
32+
epochs: 12
33+
freeze_encoder: false
34+
attack_strength: medium
35+
hard_negative:
36+
enabled: true
37+
epochs: 18
38+
freeze_encoder: true
39+
attack_strength: medium
40+
freeze_decoder: false
41+
encoder_teacher_weight: 150.0
42+
encoder_teacher_scale: 0.034
43+
loss:
44+
payload_weight: 14.0
45+
image_weight: 0.0
46+
tv_weight: 0.0
47+
residual_weight: 0.0
48+
use_fixed_residual: false
49+
finalize:
50+
enabled: true
51+
epochs: 8
52+
freeze_encoder: false
53+
attack_strength: medium
54+
freeze_decoder: false
55+
loss:
56+
payload_weight: 12.0
57+
image_weight: 0.65
58+
tv_weight: 0.018
59+
residual_weight: 0.025
60+
attack:
61+
ops_per_sample:
62+
min: 1
63+
max: 2
64+
jpeg_quality:
65+
- 75
66+
- 95
67+
webp_quality:
68+
- 75
69+
- 95
70+
resize_scale:
71+
- 0.85
72+
- 1.15
73+
crop_keep:
74+
- 0.96
75+
- 1.0
76+
rotation_deg:
77+
- -1.0
78+
- 1.0
79+
perspective_ratio:
80+
- 0.0
81+
- 0.01
82+
gaussian_blur_sigma:
83+
- 0.0
84+
- 0.35
85+
motion_blur_ksize:
86+
- 3
87+
- 5
88+
gaussian_noise_std:
89+
- 0.0
90+
- 0.004
91+
overlay_area:
92+
- 0.01
93+
- 0.04
94+
overlay_alpha:
95+
- 0.1
96+
- 0.25
97+
model:
98+
residual_scale: 0.058823529411764705
99+
dashboard:
100+
status_interval_batches: 1
101+
sample_interval_batches: 10
102+
checkpoint: artifacts/mlwm_v1/runs/20260426T174908+0000_e7ba793/best.ckpt

‎artifacts/promoted/mlwm-v1-alpha1/metrics_epoch.csv‎

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
epoch,stage,train_loss,val_payload_acc,val_exact_match,val_decode_success,val_confidence
2+
1,hard_negative,0.04573048144405491,0.9969411375661376,0.9252645502645502,0.9748677248677249,0.9964011065543644
3+
2,hard_negative,0.04366658433006159,0.9984886532738095,0.9484126984126984,0.9920634920634921,0.9955250781679911
4+
3,hard_negative,0.04120893713604253,0.9984343998015873,0.9484126984126984,0.9821428571428571,0.995825469493866
5+
4,hard_negative,0.04206904746458354,0.9988813450727514,0.9292328042328042,0.9880952380952381,0.9968866999187167
6+
5,hard_negative,0.033198884960146446,0.9980856274801587,0.9444444444444444,0.9801587301587301,0.9973790721287803
7+
6,hard_negative,0.039924131055661366,0.9991784474206349,0.9523809523809523,0.9880952380952381,0.9982844401919653
8+
7,hard_negative,0.03050802234401075,0.9990079365079365,0.9543650793650794,0.9940476190476191,0.9980981936530461
9+
8,hard_negative,0.037945131809504144,0.9988064236111112,0.9503968253968254,0.9940476190476191,0.9973026154533265
10+
9,hard_negative,0.034928924263667456,0.9984654017857143,0.9682539682539683,0.9900793650793651,0.996932998536125
11+
10,hard_negative,0.03208624450338653,0.9985816592261905,0.9464285714285714,0.9861111111111112,0.996999358373975
12+
11,hard_negative,0.02782364934266197,0.9987754216269841,0.9662698412698413,0.9940476190476191,0.9967201749483744
13+
12,hard_negative,0.03180287201591232,0.9986979166666666,0.9623015873015873,0.9841269841269841,0.9971964699881417
14+
13,hard_negative,0.028288940866040453,0.9991784474206349,0.9563492063492064,0.996031746031746,0.9974935612981282
15+
14,hard_negative,0.03839817073253836,0.9976515997023809,0.9464285714285714,0.9761904761904762,0.9963689068007091
16+
15,hard_negative,0.031832219172813404,0.9983801463293651,0.9689153439153438,0.9847883597883598,0.9973160879952567
17+
16,hard_negative,0.03439752528616459,0.9992171999007936,0.9623015873015873,0.9900793650793651,0.9973890128589812
18+
17,hard_negative,0.025241811088225617,0.9988451760912699,0.9603174603174603,0.9920634920634921,0.9963471104228308
19+
18,hard_negative,0.027990431887515124,0.9993489583333334,0.9722222222222222,0.996031746031746,0.9977506436998882
20+
19,finalize,0.06913595193409346,0.9985584077380952,0.9523809523809523,0.9920634920634921,0.9966789077198694
21+
20,finalize,0.07437173696151816,0.9976593501984127,0.9292328042328042,0.9828042328042328,0.9965127025331769
22+
21,finalize,0.06458439488368982,0.9977988591269841,0.9305555555555556,0.9821428571428571,0.9975346516049097
23+
22,finalize,0.07922951265200043,0.9988994295634921,0.9226190476190477,0.9880952380952381,0.9969448740520175
24+
23,finalize,0.07797464108137736,0.9987289186507936,0.9384920634920635,0.9880952380952381,0.9966847508672684
25+
24,finalize,0.09515007382753486,0.9977678571428571,0.9226190476190477,0.9761904761904762,0.9962382089524042
26+
25,finalize,0.09965920207919071,0.9975301752645502,0.9153439153439153,0.9808201058201058,0.9962732158009968

‎artifacts/promoted/mlwm-v1-alpha1/train_config_resolved.yaml‎

Lines changed: 106 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,106 @@
1+
seed: 20260426
2+
image_size: 448
3+
batch_size: 8
4+
grad_accum: 2
5+
num_workers: 12
6+
device: cuda
7+
amp: true
8+
max_text_bytes: 16
9+
payload_bits: 256
10+
train_dir: data/train_images
11+
val_dir: data/val_images
12+
artifacts_dir: artifacts/mlwm_v1
13+
promoted_dir: artifacts/promoted
14+
models_dir: resources/models/neural_wm
15+
optimizer:
16+
lr: 4.0e-05
17+
weight_decay: 1.0e-05
18+
loss:
19+
payload_weight: 10.0
20+
image_weight: 0.6
21+
tv_weight: 0.01
22+
residual_weight: 0.02
23+
stages:
24+
warmup:
25+
enabled: false
26+
epochs: 3
27+
freeze_encoder: true
28+
fixed_residual_scale: 0.04
29+
attack_strength: medium
30+
main:
31+
enabled: false
32+
epochs: 12
33+
freeze_encoder: false
34+
attack_strength: medium
35+
hard_negative:
36+
enabled: true
37+
epochs: 18
38+
freeze_encoder: true
39+
attack_strength: medium
40+
freeze_decoder: false
41+
encoder_teacher_weight: 150.0
42+
encoder_teacher_scale: 0.034
43+
loss:
44+
payload_weight: 14.0
45+
image_weight: 0.0
46+
tv_weight: 0.0
47+
residual_weight: 0.0
48+
use_fixed_residual: false
49+
finalize:
50+
enabled: true
51+
epochs: 8
52+
freeze_encoder: false
53+
attack_strength: medium
54+
freeze_decoder: false
55+
loss:
56+
payload_weight: 12.0
57+
image_weight: 0.65
58+
tv_weight: 0.018
59+
residual_weight: 0.025
60+
attack:
61+
ops_per_sample:
62+
min: 1
63+
max: 1
64+
jpeg_quality:
65+
- 75
66+
- 95
67+
webp_quality:
68+
- 75
69+
- 95
70+
resize_scale:
71+
- 0.85
72+
- 1.15
73+
crop_keep:
74+
- 0.96
75+
- 1.0
76+
rotation_deg:
77+
- -1.0
78+
- 1.0
79+
perspective_ratio:
80+
- 0.0
81+
- 0.01
82+
gaussian_blur_sigma:
83+
- 0.0
84+
- 0.35
85+
motion_blur_ksize:
86+
- 3
87+
- 5
88+
gaussian_noise_std:
89+
- 0.0
90+
- 0.004
91+
overlay_area:
92+
- 0.01
93+
- 0.04
94+
overlay_alpha:
95+
- 0.1
96+
- 0.25
97+
enabled_ops:
98+
- crop
99+
- screenshot_sim
100+
- color_jitter
101+
model:
102+
residual_scale: 0.058823529411764705
103+
dashboard:
104+
status_interval_batches: 1
105+
sample_interval_batches: 10
106+
checkpoint: artifacts/mlwm_v1/runs/20260426T174908+0000_e7ba793/best.ckpt

0 commit comments

Comments
 (0)