模型上线

train/train_service_sk_tf2_distibute.py

@@ -89,7 +89,7 @@ def input_fn(csv_path, epoch, shuffle, batch_size):
 
     dataset = dataset.map(parse_line, num_parallel_calls = 8).cache()
     dataset = dataset.padded_batch(batch_size, padded_shapes, padding_values=padding_values)
-    dataset = dataset.prefetch(2048 * 5).repeat(epoch)
+    dataset = dataset.prefetch(512 * 100).repeat(epoch)
     if shuffle:
         dataset.shuffle(2048)