Think about storing the training data as tf records. Will it fasten the training stage? related to #96