plot digits from validdation set
merge conflict and plot digits from valid set