xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

set_train_false 

Signed-off-by: zhupuxu <zhupuxu@huawei.com>
This commit is contained in:
zhupuxu 2020-12-24 16:09:02 +08:00
parent 4df0baa6c7
commit 33dc7d3755
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
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)