I am newbie studying neural network using pytorch, now I am trying to recognize a mushroom image whether it is poisonous or not, by comparing it with training labels.