/nn_utils

Collection of some useful losses.

Primary LanguageJupyter Notebook

Loss Collection

Associative Embedding Loss

Focal Loss