in sparse_autoencoder/train.py [0:0]
def logkv(self, k, v): if self.enabled: self.vals[k] = v.detach() if isinstance(v, torch.Tensor) else v return v