From a36c68b1ab0affbcdbfd592b241723f7818723a3 Mon Sep 17 00:00:00 2001 From: yao_yf Date: Sat, 29 Aug 2020 09:41:55 +0800 Subject: [PATCH] no_bool_parser_in_wide_and_deep_multitable --- .../official/recommend/wide_and_deep_multitable/src/config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model_zoo/official/recommend/wide_and_deep_multitable/src/config.py b/model_zoo/official/recommend/wide_and_deep_multitable/src/config.py index 2e4005c06e..df066119d2 100644 --- a/model_zoo/official/recommend/wide_and_deep_multitable/src/config.py +++ b/model_zoo/official/recommend/wide_and_deep_multitable/src/config.py @@ -31,7 +31,7 @@ def argparse_init(): parser.add_argument("--adam_lr", type=float, default=0.003) # The Adam lr parser.add_argument("--ftrl_lr", type=float, default=0.1) # The ftrl lr. parser.add_argument("--l2_coef", type=float, default=0.0) # The l2 coefficient. - parser.add_argument("--is_tf_dataset", type=bool, default=True) # The l2 coefficient. + parser.add_argument("--is_tf_dataset", type=int, default=1) # The l2 coefficient. parser.add_argument("--dropout_flag", type=int, default=1) # The dropout rate parser.add_argument("--output_path", type=str, default="./output/") # The location of the output file. @@ -87,7 +87,7 @@ class WideDeepConfig(): self.l2_coef = args.l2_coef self.ftrl_lr = args.ftrl_lr self.adam_lr = args.adam_lr - self.is_tf_dataset = args.is_tf_dataset + self.is_tf_dataset = bool(args.is_tf_dataset) self.output_path = args.output_path self.eval_file_name = args.eval_file_name