Fig. 5: Schematic diagram of CNN model architecture. | npj Computational Materials