A modular graph based DataSet implementation for Pytorch - View it on GitHub