Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

set_train_false 

Signed-off-by: zhupuxu <zhupuxu@huawei.com>
tags/v1.1.0
zhupuxu 5 years ago
parent
fdde3b75c6
commit
4e0567cdb7
1 changed files with 1 additions and 0 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  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) model_builder = ModelBuilder(ModelConfig, TrainConfig)
_, network = model_builder.get_train_eval_net() _, network = model_builder.get_train_eval_net()
network.set_train(False)


load_checkpoint(args.ckpt_file, net=network) load_checkpoint(args.ckpt_file, net=network)




Write Preview
Loading…
Cancel
Save