I want to perform segmentation of optic disc from fundus images using attention networks. The architecture of the model is picked from "https://github.com/lixiaolei1982