diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-04-07 23:10:56 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-04-07 23:10:56 -0500 |
| commit | 89fff0048c04bdc4c8beb6d11f8d5564d75cbb0c (patch) | |
| tree | 97048ea3de1ae0662b82e87a7363a3a16e11fe60 /metrics/__pycache__/credit_metrics.cpython-313.pyc | |
| parent | 7fbbe2c18a08f0a6314dfe22dc8790462252050a (diff) | |
Add ViT-Mini DFA training script that saves checkpoint + Bs
The existing snapshot_evolution_vit.py and vit_frozen_blocks_baseline.py
do not save model checkpoints — they only emit per-epoch JSON logs. This
makes it impossible to apply the diagnostic protocol to a trained ViT
post-hoc, since the protocol needs an actual model object.
This script trains a 4-block d=128 ViT-Mini with block-level DFA on
CIFAR-10 (same training rule as snapshot_evolution_vit.py) for 60 epochs
and saves:
- the final state_dict
- the random feedback Bs (so the protocol can also verify bug 4 on
this checkpoint)
- test_acc and config
Output: results/vit_dfa_checkpoints/dfa_vit_s{seed}.pt
Diffstat (limited to 'metrics/__pycache__/credit_metrics.cpython-313.pyc')
0 files changed, 0 insertions, 0 deletions
