In addition to class balanced losses, this repo also supports the standard versions of the cross entropy/focal loss etc. over the same API All typos and errors in vandit15's source are fixed ...