Cnn Model Python For Loop