Generalized the DataGenerator to accept two lists of input and output (wrt network) labels to fetch from the dataset files. ca253db jpdefrutos commited on Jan 31, 2021