Currently, loss function is only allowed to receive network's output and ground-truth.
To increase the flexibility of loss function, additional parameters should also be visible to loss function.
Solution:
Add additional key like additional_loss_data to pass the additional data to loss function.