模型上线

0b82955c · 宋柯 · c3d92073 · 0b82955c
Commit 0b82955c authored 3 years ago by 宋柯
Hide 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', 3, False, 2048), hooks = hooks)
+train_spec = tf.estimator.TrainSpec(input_fn = lambda: input_fn(DATA_DIR + 'train_samples.csv', 3, True, 2048), hooks = hooks)

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