ldist_loss not defined in main code

[Shashanka Venkataramanan]

Thank you for providing this code.
Actually, ldist_loss has not been defined in main.py.
If it has been a mistake from my end, kindly forgive me for the same. Thanks

[Jamie Hayes]

Thanks, should be fixed now