Fix demo of pruning. (#79)

ae7ff429 · whs · GitHub · 5a8f87af · ae7ff429 · ae7ff429
隐藏空白更改
内联并排

Showing with 3 addition and 2 deletion

demo/prune/train.py demo/prune/train.py +2 -2

requirements.txt requirements.txt +1 -0

未找到文件。
--- a/demo/prune/train.py
+++ b/demo/prune/train.py
@@ -208,7 +208,7 @@ def compress(args):
        val_program,
        fluid.global_scope(),
        params=params,
-        ratios=[FLAGS.pruned_ratio] * len(params),
+        ratios=[args.pruned_ratio] * len(params),
        place=place,
        only_graph=True)

@@ -216,7 +216,7 @@ def compress(args):
        fluid.default_main_program(),
        fluid.global_scope(),
        params=params,
-        ratios=[FLAGS.pruned_ratio] * len(params),
+        ratios=[args.pruned_ratio] * len(params),
        place=place)
    _logger.info("FLOPs after pruning: {}".format(flops(pruned_program)))
    for i in range(args.num_epochs):

--- a/requirements.txt
+++ b/requirements.txt
 #paddlepaddle == 1.6.0rc0
+tqdm