def iter_dataset(dset):
    """
    Necessary because native pytorch implementation of __iter__ on Dataset is problematic
    :param dset:
    :return:
    """
    for i in range(len(dset)):
        yield dset[i]
