support saving model r2
add linear ae
fix stl