diff --git a/elk/training/ccs_reporter.py b/elk/training/ccs_reporter.py index 29738dd0..3193001e 100644 --- a/elk/training/ccs_reporter.py +++ b/elk/training/ccs_reporter.py @@ -100,7 +100,7 @@ def __init__( self.norm = None - if self.cfg.norm == "burns": + if self.config.norm == "burns": self.probe = nn.Sequential( nn.Linear( in_features,