olegboev
Results
1
comments of
olegboev
The problem is that the loss value is a vector of batch size, so it is required to calculate its mean. This fragment: ``` return { "loss": total_loss, "reconstruction_loss": reconstruction_loss,...