Fix size in convnet comment