It trained well, and gave great results! Thanks for the reply, it was a super silly mistake because I was using the loss function the wrong way. Figured it out after spending some time on the docs  Thank you!