Thanks @Rohan_Kumar for looking into it. This is how the loss looks like over epochs.
epoch1: 214.88601249456406
epoch2: 214.87873661518097
epoch3: 214.88536673784256
epoch4: 214.88878732919693
epoch5: 214.8846591114998
epoch6: 214.882130920887
epoch7: 214.87578403949738
epoch8: 214.8743262887001
epoch9: 214.90140038728714
epoch10: 214.8920682668686
epoch11: 214.86986935138702
epoch12: 214.89466083049774
epoch13: 214.88981753587723
epoch14: 214.88611370325089
epoch15: 214.89879924058914