Hands-On GPU Computing with Python

PyTorch output on a PyCharm console

First, we display the GPU device name:

Device used: GeForce GTX TITAN XFiles already downloaded and verifiedFiles already downloaded and verified

As we will see next, we can identify a ship, a bird, a frog, and another bird. These are pre-trained images, so the labels are shown correctly:

ship bird frog bird[1, 2000] loss: 2.161[1, 4000] loss: 1.832[1, 6000] loss: 1.663[1, 8000] loss: 1.589[1, 10000] loss: 1.519[1, 12000] loss: 1.455[2, 2000] loss: 1.381[2, 4000] loss: 1.357[2, 6000] loss: 1.335[2, 8000] loss: 1.324[2, 10000] loss: 1.313[2, 12000] loss: 1.281Finished Training

After training, we first show the correct prediction and cross-validate it with the new prediction. Here, we see that apart from ...