Solved by BlueTurtle in post #2 
              
              
                This problem was solved with: 
            images = torch.FloatTensor(images[0])
            images = images.unsqueeze(1)
            images = images.permute(3, 0, 1, 2)

I am still having problems with this notebook for other reasons but this has been solved.