From 8389593cfcfde3ffc156109f6a24a14ed5e36381 Mon Sep 17 00:00:00 2001 From: Pawan KS Date: Sat, 13 Feb 2021 04:58:52 +0530 Subject: [PATCH] Update pruning.py changed 54th line --- pruning.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pruning.py b/pruning.py index edb3993..60b9460 100755 --- a/pruning.py +++ b/pruning.py @@ -51,7 +51,7 @@ model = get_model(args.model, 'prune', data_object.num_classes, data_object.insize) # state = torch.load(f"checkpoints/{args.model}_{args.dataset}_pretrained.pth") # state = torch.load("../resnet152-b121ed2d.pth") -# model.load_state_dict(state['state_dict'], strict=False) +# model.load_state_dict(state, strict=False) ############################### preparing for pruning ################################### @@ -187,4 +187,4 @@ def test(model, loss_fn, optimizer, phase, epoch): df_data=np.array([remaining_before_pruning, remaining_after_pruning, valid_accuracy, pruning_accuracy, pruning_threshold, problems]).T df = pd.DataFrame(df_data,columns = ['Remaining before pruning', 'Remaining after pruning', 'Valid accuracy', 'Pruning accuracy', 'Pruning threshold', 'problems']) - df.to_csv(f"logs/{name}.csv") \ No newline at end of file + df.to_csv(f"logs/{name}.csv")