diff --git a/distiller/thinning.py b/distiller/thinning.py index 58ceb1c20b31372550488521b50c19a3cb0bca1a..434668746fa848f633739fc0450d0c788ea025d9 100755 --- a/distiller/thinning.py +++ b/distiller/thinning.py @@ -206,7 +206,7 @@ def find_nonzero_channels_list(param, param_name): def apply_and_save_recipe(model, zeros_mask_dict, thinning_recipe, optimizer): if len(thinning_recipe.modules) > 0 or len(thinning_recipe.parameters) > 0: - # Now actually remove the filters, chaneels and make the weight tensors smaller + # Now actually remove the filters, channels and make the weight tensors smaller execute_thinning_recipe(model, zeros_mask_dict, thinning_recipe, optimizer) # Stash the recipe, so that it will be serialized together with the model