fixes multiclass training as done in #193, including refactoring of the MetricsConfig
still enables the use of models before the refactoring with 71c20301a97728fc019704712a42f3a62bf58fbb
removes unused parameters label_counts and class_weights_loss
I tested the code on an efficientnet trained in DT for HSN and with pretrained models of eat (see other PR).
Please check if you are able to run your models.
This PR:
MetricsConfig
label_counts
andclass_weights_loss
I tested the code on an efficientnet trained in DT for HSN and with pretrained models of eat (see other PR). Please check if you are able to run your models.