Max Prod

Results 1 issues of Max Prod

In the current implementation, the `forward()` method is generic for train or eval mode. In some case, we need to have not only the loss but the prediction on output...