@albanD can you help me out with this. When i start training with values sx, sq = 0.0, -3.0 as suggested by the author my loss is diverging to negative.
epoch1-------[02:56<00:00,  3.62it/s, average_loss=-18.351]
epoch2-------[02:54<00:00,  3.66it/s, average_loss=-22.009]
epoch3-------[02:54<00:00, 4.55it/s, average_loss=-23.185]


Is it how it is supposed to work?