Pytorch-image-classification Almost any image classification task in pytorch To do Reduce lr on plateau Learning rate scheduler Tensorboard Number of elements in each class plot Change print function