fix prune optimer

PaddlePaddle · Apr 28, 2022 · f83524b · f83524b
1 parent f5f8af4
commit f83524b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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'])