function not defined

[draymondLiu]

Could you provide the code of 'load_model_and_prune' and 'load_model' in test.py?

[dengbaowang]

Same request.

[dengbaowang]

Could you provide the code of 'load_model_and_prune' and 'load_model' in test.py?

Hi Draymond, I found that the function "build_model" shown in test_prune.py is actually your expected function "load_model_and_prune". So you can just change the function name to "build_model" in line 193. Also, load_model(model_path) could be replaced by model.load_state_dict(torch.load(model_path)). Hope this help you :)