Fix predict_SAE bug in data feeding