Skip to content

feat: add an encoder/decoder with a memory bank#153

Open
vijk777 wants to merge 3 commits intomainfrom
vj/z0_enc_dec
Open

feat: add an encoder/decoder with a memory bank#153
vijk777 wants to merge 3 commits intomainfrom
vj/z0_enc_dec

Conversation

@vijk777
Copy link
Collaborator

@vijk777 vijk777 commented Feb 25, 2026

Summary

  • Adds zapbench_z0_enc_dec.py training script for sparse z0 bank + linear encoder/decoder
  • Phase 1: decoder training (z0_bank + decoder)
  • Phase 2: encoder training to match z0_bank targets
  • Generates diagnostic plots (L2 distance, t-SNE) for train/val data
  • Supports --max_conditions N for fast testing

To run a quick test:

python -m LatentEvolution.zapbench_z0_enc_dec test_fast \
  --decoder_epochs 2 --enc_epochs 2       \
  --max_conditions 1

🤖 Generated with Claude Code

vijk777 and others added 3 commits February 20, 2026 16:34
trains sparse z0 bank with linear encoder/decoder on zapbench data:
- phase 1: decoder training (z0_bank + decoder)
- phase 2: encoder training to match z0_bank targets
- generates diagnostic plots (L2 distance, t-SNE)
- supports max_conditions option for fast testing

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant