[loss] focal loss throwing an error when used in place of ce loss

NVIDIA / VariantWorks

Deep Learning based variant calling toolkit - https://clara-parabricks.github.io/VariantWorks/

Apache License 2.0

44 stars 11 forks source link

[loss] focal loss throwing an error when used in place of ce loss #158

Open tijyojwad opened 3 years ago

tijyojwad commented 3 years ago

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