* Add InfiniteDataLoader Only initializes at first epoch. Saves time. * Moved class to a better location