update medium mnist graph