TODO: tune SZAE and adjust the LR layer.