Fig. 2: The architecture of attention U-net. | npj Computational Materials