def get_one_batch(dataloader):
    """Returns one batch from a dataloader"""
    iter_dl = iter(dataloader) #Necessary.  You have to tell the fucking thing it's iterable.  Why?
    batch = next(iter_dl)
    return(batch)