diff --git a/docs/zh_cn/quick_start/pruning_tutorial.md b/docs/zh_cn/quick_start/pruning_tutorial.md index 051a740f4602f2175537efe5eaf22b62a06c10e7..d34bb9ce48a8ca80bd37d00d640ea2a6ad669ed1 100755 --- a/docs/zh_cn/quick_start/pruning_tutorial.md +++ b/docs/zh_cn/quick_start/pruning_tutorial.md @@ -61,7 +61,7 @@ pruned_program, _, _ = pruner.prune( ### 3.3 计算剪裁之后的FLOPs ``` -FLOPs = paddleslim.analysis.flops(train_program) +FLOPs = paddleslim.analysis.flops(pruned_program) print("FLOPs: {}".format(FLOPs)) ```