search

soniajoseph / ViT-Prisma

ViT Prisma is a mechanistic interpretability library for Vision Transformers (ViTs).
Other
165 stars 18 forks source link

Clean up configuration for SAE training #94

Open themachinefan opened 3 months ago

themachinefan commented 3 months ago

Replace the hacky argparse and hardcoded config parameters with nicer yaml input (or equivalent)

soniajoseph commented 3 months ago

(likely inspired by SAELens)

PraneetNeuro commented 3 months ago

Picking up this issue