This is a PyTorch template for classification, integrating focal loss and tensorboard visualization.
run python train.py to train your model, you can find your model and training logs under the folder logs/
run tensorboard --logdir runs to visualize your training process
You can also see the accuracy of each class during training and testing
