From e5afb3394e7034482927e4155e3f9eddbbf78851 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Fri, 22 Dec 2023 07:58:04 +0100 Subject: analyze-log: add --show-model-size support --- bin/analyze-log.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'bin') diff --git a/bin/analyze-log.py b/bin/analyze-log.py index 4c37928..2192a2e 100755 --- a/bin/analyze-log.py +++ b/bin/analyze-log.py @@ -267,6 +267,9 @@ def main(): error_metric=args.error_metric, ) + if args.show_model_size: + dfatool.cli.print_model_size(model) + if args.export_model: print(f"Exportding model to {args.export_model}") json_model = model.to_json() -- cgit v1.2.3