JasonGross / guarantees-based-mechanistic-interpretability

guarantees-based-mechanistic-interpretability/notebooks_jason/max_of_2_grokking.py

Lines 126 to 128 in d0cf251

    
           models = runtime.model_versions(cfg, max_count=3000, step=1) 
        
           assert models is not None 
        
           models = list(models)

re-downloads the artifacts from wandb every time it is run. It'd be nice to cache them on disk somehow. (Does wandb support this?)

Alternatively (perhaps better) we need to convince someone to fix wandb/wandb#7006

	models = runtime.model_versions(cfg, max_count=3000, step=1)
	assert models is not None
	models = list(models)

wandb downloads should be cached