From f83524bdc5d0c295861bd79f35fb02ecc55f5b19 Mon Sep 17 00:00:00 2001
From: Topdu <784990967@qq.com>
Date: Thu, 28 Apr 2022 09:00:41 +0000
Subject: [PATCH] fix prune optimer
---
deploy/slim/prune/sensitivity_anal.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/deploy/slim/prune/sensitivity_anal.py b/deploy/slim/prune/sensitivity_anal.py
index 306f1a83a..be64a6bcf 100644
--- a/deploy/slim/prune/sensitivity_anal.py
+++ b/deploy/slim/prune/sensitivity_anal.py
@@ -94,7 +94,7 @@ def main(config, device, logger, vdl_writer):
config['Optimizer'],
epochs=config['Global']['epoch_num'],
step_each_epoch=len(train_dataloader),
- parameters=model.parameters())
+ model=model)
# build metric
eval_class = build_metric(config['Metric'])