Multi loss - Githubissues

MathGaron / pytorch_toolbox

Boiler plate code for pytorch. Train/Validation loops, visualization etc. For research.

MIT License

10 stars 3 forks source link

Open MathGaron opened 4 years ago

MathGaron commented 4 years ago

It is now possible to return a dict from the network loss method:

We should add this feature to the log system (so that we don't have to recompute them in the callbacks)
We should document this at least in the examples. (Now we have the old behavior with an int + the dict)