模型调试

train/train_service_sk.py

@@ -78,7 +78,7 @@ 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)
+    dataset = dataset.map(parse_line, num_parallel_calls=8).cache()
     dataset = dataset.padded_batch(batch_size, padded_shapes, padding_values=padding_values)
     if shuffle:
         dataset = dataset.shuffle(1000).prefetch(512 * 100).repeat(epoch)