From 33dc7d37551f06b7c8385a8920a26dd340931934 Mon Sep 17 00:00:00 2001 From: zhupuxu Date: Thu, 24 Dec 2020 16:09:02 +0800 Subject: [PATCH] set_train_false Signed-off-by: zhupuxu --- model_zoo/official/recommend/deepfm/export.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model_zoo/official/recommend/deepfm/export.py b/model_zoo/official/recommend/deepfm/export.py index 87c38aafc98..230bafa75b3 100644 --- a/model_zoo/official/recommend/deepfm/export.py +++ b/model_zoo/official/recommend/deepfm/export.py @@ -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)