fix prune optimer

2022-04-28 09:00:41 +00:00 · 2022-04-28 09:00:41 +00:00 · f83524bdc5
parent f5f8af4dad
commit f83524bdc5
1 changed files with 1 additions and 1 deletions
--- 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'])