模型上线

7037767c · 宋柯 · 9a1f6416 · 7037767c
Commit 7037767c authored Dec 30, 2021 by 宋柯
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

train_service_sk_tf2_distibute.py train/train_service_sk_tf2_distibute.py +1 -1

No files found.
--- a/train/train_service_sk_tf2_distibute.py
+++ b/train/train_service_sk_tf2_distibute.py
@@ -237,7 +237,7 @@ early_stopping = tf.estimator.experimental.stop_if_no_increase_hook(wideAndDeepM

 hooks = [early_stopping]

-train_spec = tf.estimator.TrainSpec(input_fn = lambda: input_fn(DATA_DIR + 'train_samples.csv', 1, False, 2048), hooks = hooks)
+train_spec = tf.estimator.TrainSpec(input_fn = lambda: input_fn(DATA_DIR + 'train_samples.csv', 3, False, 2048), hooks = hooks)

 serving_feature_spec = tf.feature_column.make_parse_example_spec(
  linear_feature_columns + dnn_feature_columns)