Our RMSE on Movielens is strange compared to SOTA: https://paperswithcode.com/sota/collaborative-filtering-on-movielens-100k There may be a bug. Or we may be plotting the train RMSE.