Skip to content

Commit 354b077

Browse files
kuizhiqingroot
authored andcommitted
Update tfrecord_writer.py
1 parent e9beb75 commit 354b077

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/dataset/utils/tfrecord_writer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def serialize_example(dense_feature, sparse_feature, label):
5858
if __name__ == '__main__':
5959

6060
output_file = 'wd.tfrecord'
61-
input_file = '/work/FleetX/examples/wide_and_deep/data/part-0'
61+
input_file = './part-0'
6262

6363
# setting for this dataset only
6464
continuous_range_ = range(1, 14)

0 commit comments

Comments
 (0)