I’m referring to your second guess: how can I differentiate between classes that constitute an NN model and those which implement losses …
I want to write a python function to do that for me and clearly basing on class names is not a good way to do it.
Anyway, thank you very much for your help.