未验证 提交 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:
model_filename: model.pdmodel
params_filename: model.pdiparams
deploy_hardware: SD710
input_shapes: [1,3,398,224]
TrainConfig:
epochs: 14
......
......@@ -164,7 +164,8 @@ def main(args):
config=all_config,
train_dataloader=train_dataloader,
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
ac.compress()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册