模型上线

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)