Skip to content

Commit f6cd9db

Browse files
CEQ151CEQ151
authored andcommitted
feat(mlwm-train): add main-only decode candidate config
1 parent 6107b38 commit f6cd9db

1 file changed

Lines changed: 26 additions & 0 deletions

File tree

configs/mlwm/main_only_5090.yaml

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
extends: configs/mlwm/bootstrap_5090.yaml
2+
stages:
3+
warmup:
4+
enabled: true
5+
epochs: 3
6+
freeze_encoder: true
7+
use_fixed_residual: true
8+
fixed_residual_scale: 0.04
9+
attack_strength: clean
10+
main:
11+
enabled: true
12+
epochs: 12
13+
freeze_encoder: false
14+
freeze_decoder: true
15+
encoder_teacher_weight: 20000.0
16+
encoder_teacher_scale: 0.04
17+
attack_strength: clean
18+
loss:
19+
payload_weight: 1.0
20+
image_weight: 0.0
21+
tv_weight: 0.0
22+
residual_weight: 0.0
23+
hard_negative:
24+
enabled: false
25+
finalize:
26+
enabled: false

0 commit comments

Comments
 (0)