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)