diff --git a/apputils/checkpoint.py b/apputils/checkpoint.py index 10c1ed97b2b98752bce500cbb8e99a9cb441f122..e6fc056b72f19c7f2c275a17fcfc637109b4bfd4 100755 --- a/apputils/checkpoint.py +++ b/apputils/checkpoint.py @@ -107,8 +107,8 @@ def load_checkpoint(model, chkpt_file, optimizer=None): # Cache the recipes in case we need them later model.thinning_recipes = checkpoint['thinning_recipes'] distiller.execute_thinning_recipes_list(model, - compression_scheduler.zeros_mask_dict, - model.thinning_recipes) + compression_scheduler.zeros_mask_dict, + model.thinning_recipes) if 'quantizer_metadata' in checkpoint: msglogger.info('Loaded quantizer metadata from the checkpoint')