Update prune.py

FIX description CIFAR -> Imagenet
Eric-mingjie · Apr 23, 2020 · ed1ca16 · ed1ca16
1 parent f328cca
commit ed1ca16
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/imagenet/network-slimming/prune.py b/imagenet/network-slimming/prune.py
@@ -13,7 +13,7 @@
 
 
 # Prune settings
-parser = argparse.ArgumentParser(description='PyTorch Slimming CIFAR prune')
+parser = argparse.ArgumentParser(description='PyTorch Slimming Imagenet prune')
 parser.add_argument('--data', type=str, default='',
                     help='Path to imagenet validation data')
 parser.add_argument('--test-batch-size', type=int, default=16, metavar='N',
@@ -187,4 +187,4 @@ def update(self, val, n=1):
 acc = test()
 
 print(cfg)
-print("Accuracy after pruning top1: %f top5: %f"%(acc[0], acc[1]))
+print("Accuracy after pruning top1: %f top5: %f"%(acc[0], acc[1]))