Skip to content

v0.5.0

Choose a tag to compare

View all tags
@xtmtd xtmtd released this 10 Sep 10:17
· 11 commits to main since this release
v0.5.0
d1a807c

What's new since v0.4.0

  • ArcFace fine-tuning: new runs use a compact unnormalized embedding head (backbone_dim -> 512 -> metric_embed_dim) matching the reference script. --metric-embed-dim now actually resizes the head.
  • New options: --metric-head-lr, --weight-decay (default 1e-4; pass 0.05 for the legacy script). SFT gradient clipping removed.
  • Resume safety: checkpoints record embedding_head and freeze_ratio; --resume validates them and rejects a changed embedding dimension.
  • Checkpoint compatibility: extract/export/cam share one resolver, can read ref-script checkpoints, and gained --model-name where needed. A wrong backbone now fails with a clear message.
  • Breaking: --weight-decay default changed from 0.01 to 1e-4; 0.4.0 fine-tune checkpoints load as projection_mlp_2048; finetune cannot resume ref-script checkpoints.

441 tests pass.

Assets 2
Loading