模型上线

cf014bd2 · 宋柯 · 05d1c6b9 · cf014bd2
Commit cf014bd2 authored 3 years ago 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
@@ -247,7 +247,7 @@ def train_dataset_input_fn(train_dataset):
 def eval_dataset_input_fn(eval_dataset):
    return eval_dataset

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

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