Browse Source

!10440 fix gpu deepfm mindir load failed

From: @zhupuxu
Reviewed-by: @zhoufeng54,@kisnwang
Signed-off-by: @kisnwang
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
19ef45f1a4
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      model_zoo/official/recommend/deepfm/export.py

+ 1
- 0
model_zoo/official/recommend/deepfm/export.py View File

@@ -39,6 +39,7 @@ if __name__ == "__main__":

model_builder = ModelBuilder(ModelConfig, TrainConfig)
_, network = model_builder.get_train_eval_net()
network.set_train(False)

load_checkpoint(args.ckpt_file, net=network)



Loading…
Cancel
Save