Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

[BUG] finding leaf modules #2241

Merged
merged 15 commits into from
Mar 27, 2020
Prev Previous commit
Next Next commit
update
  • Loading branch information
QuanluZhang committed Mar 23, 2020
commit 48be5519a458447a1a31d495699270c7ea5a46a4
4 changes: 2 additions & 2 deletions examples/model_compress/model_speedup.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,8 +87,8 @@ def model_inference(config):
if args.masks_file is not None:
config[args.example_name]['masks_file'] = args.masks_file
if not os.path.exists(config[args.example_name]['masks_file']):
msg = '{} does not exist! You should specify masks_file correctly, '
msg += 'or use default one which is generated by model_prune_torch.py'
msg = '{} does not exist! You should specify masks_file correctly, ' \
'or use default one which is generated by model_prune_torch.py'
raise RuntimeError(msg.format(config[args.example_name]['masks_file']))
model_inference(config[args.example_name])
else:
Expand Down