Imported from https://github.com/cinastanbean/Pytorch-Multi-Task-Multi-class-Classification