未验证 提交 a8b487df 编写于 作者: Z zhouzj 提交者: GitHub

Use fixed-shape models for hardware-aware automatic strategy selection. (#1396)

* Use fixed-shape models for automatic strategy selection.

* Fixed error.
上级 ea6043d6
...@@ -4,6 +4,7 @@ Global: ...@@ -4,6 +4,7 @@ Global:
model_filename: model.pdmodel model_filename: model.pdmodel
params_filename: model.pdiparams params_filename: model.pdiparams
deploy_hardware: SD710 deploy_hardware: SD710
input_shapes: [1,3,398,224]
TrainConfig: TrainConfig:
epochs: 14 epochs: 14
......
...@@ -164,7 +164,8 @@ def main(args): ...@@ -164,7 +164,8 @@ def main(args):
config=all_config, config=all_config,
train_dataloader=train_dataloader, train_dataloader=train_dataloader,
eval_callback=eval_function if nranks > 1 and rank_id != 0 else None, eval_callback=eval_function if nranks > 1 and rank_id != 0 else None,
deploy_hardware=config.get('deploy_hardware') or None) deploy_hardware=config.get('deploy_hardware') or None,
input_shapes=config.get('input_shapes', None))
# step3: start the compression job # step3: start the compression job
ac.compress() ac.compress()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册