diff --git a/clu/parameter_overview.py b/clu/parameter_overview.py index a47823a..3dc9337 100644 --- a/clu/parameter_overview.py +++ b/clu/parameter_overview.py @@ -98,6 +98,13 @@ def count_parameters(params: _ParamsContainer) -> int: def _make_row(name, value) -> _ParamRow: + if value is None: + return _ParamRow( + name=name, + shape=(), + dtype="", + size=0, + ) return _ParamRow( name=name, shape=value.shape,