Fix Nested Parameters, Update Training script, and Document dSprites Experiment

soniajoseph / ViT-Prisma

ViT Prisma is a mechanistic interpretability library for Vision Transformers (ViTs).

Other

165 stars 18 forks source link

Closed YashVadi closed 10 months ago

YashVadi commented 10 months ago

Fixed the issue of nested parameters in the sweep config.
Updated the configuration to use a timestamp as the default run_id.
Updated the training source to handle the creation of the checkpoint directory if it does not exist.
Documented the results of the dSprites mice training experiment.

Todo

YashVadi commented 10 months ago

I added the model checkpoint paths for all the dSprites models.

soniajoseph commented 10 months ago

Excellent ty!