模型上线

train/train_service_sk_tf2_distibute.py

@@ -87,9 +87,9 @@ def input_fn(csv_path, epoch, shuffle, batch_size):
                        'ITEM_MULTI_CATEGORY_second_solutions': '-1', 'ITEM_MULTI_CATEGORY_second_positions': '-1',
                        'ITEM_MULTI_CATEGORY_projects': '-1', 'ITEM_NUMERIC_sku_price': 0.0}, 0.0)
 
-    dataset = dataset.map(parse_line, num_parallel_calls = 8).cache()
+    dataset = dataset.prefetch(512 * 100).map(parse_line, num_parallel_calls = 8).cache()
     dataset = dataset.padded_batch(batch_size, padded_shapes, padding_values=padding_values)
-    dataset = dataset.prefetch(512 * 100).repeat(epoch)
+    dataset = dataset.repeat(epoch)
     if shuffle:
         dataset.shuffle(2048)