There is no pretrained method fot batchtopk in dictionary.py

[Noam-Diamant]

hi.
i have trained a batchtopk sae, but i am unable to load it since there is no batchtopk class in dictionary.py with pretrained method.
is there any way to load my trained sae ?

[adamkarvonen]

https://github.com/saprmarks/dictionary_learning/blob/main/dictionary_learning/trainers/batch_top_k.py#L17

The topk saes are in the trainer class, which is unfortunately a break from the typical convention in this repo

[Noam-Diamant]

@adamkarvonen
I think I will able to fix this and PR sometime in the near future.
in the meantime, I added the relevant method from utils.py that will make a quiick fix to this problem.