fix recovery_learning_rate

qfgaohao · Jun 1, 2018 · 1a35563 · 1a35563
1 parent 49f0b67
commit 1a35563
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prune_alexnet.py b/prune_alexnet.py
@@ -26,7 +26,7 @@
                     help='number of batches to train')
 parser.add_argument('--num_recovery_batches', default=2, type=int,
                     help='number of batches to train to recover the network')
-parser.add_argument('--recovery_learning_rate', default=1e-4, type=int,
+parser.add_argument('--recovery_learning_rate', default=1e-4, type=float,
                     help='learning rate to recover the network')
 parser.add_argument('--recovery_batch_size', default=32, type=int,
                     help='Batch size for training')