[loss] focal loss throwing an error when used in place of ce loss #158
Description
When the nemo cross entropy loss is replaced with the focal loss call, it throws pytorch errors. The focal loss module should be s drop in replacement for the ce loss
{{ message }}