Merge branch 'main' into light-reporter-saving

elk/training/eigen_reporter.py

@@ -29,6 +29,9 @@ class EigenReporterConfig(ReporterConfig):
  """Whether to save the reporter statistics to disk in EigenReporter.save(). This
  is useful for debugging and analysis, but can take up a lot of disk space."""
 
+ use_centroids: bool = True
+ """Whether to average hiddens within each cluster before computing covariance."""
+
  def __post_init__(self):
  if not (0 <= self.neg_cov_weight <= 1):
  raise ValueError("neg_cov_weight must be in [0, 1]")