likely unstable due to small batch size [[ 1397 541] [ 1140 34102]] 0.7208462332301342 0.9676522331309233