invest pytorch mnist cnn videota (228)