diff --git a/examples/classifier_compression/compress_classifier.py b/examples/classifier_compression/compress_classifier.py index cc7965e717a06514530f45ae2859ab5d09ccd040..883ff9285a9ca68b01438155a1d1cbac12811fa6 100755 --- a/examples/classifier_compression/compress_classifier.py +++ b/examples/classifier_compression/compress_classifier.py @@ -196,8 +196,8 @@ def main(): # This sample application can be invoked to produce various summary reports. if args.summary: - for s in which_summary: - distiller.model_summary(model, s, dataset) + for summary in args.summary: + distiller.model_summary(model, summary, args.dataset) return if args.export_onnx is not None: