How to use your own data set for training or testing? Do you need to convert the dataset into tf.records file?