Imported from https://github.com/ppriyank/-Online-Soft-Mining-and-Class-Aware-Attention-Pytorch