Solucionado
Para evitar a divisão por 255 todas vez que a célula for executada, uma outra alternativa é atribuir uma nova variável para as imagens de treino normalizadas.
imagens_treino_norm = imagens_treino / float(255)
Dessa forma não é mais necessário dividir o código em duas células diferentes, e você não "descarta" as imagens de treino "brutas" sobrescrevendo a mesma variável.
Olá Gustavo, tudo bem?
Sua sugestão é realmente muito interessante e útil! A ideia de criar uma nova variável imagens_treino_norm para armazenar as imagens de treino é ótima alternativa para evitar a divisão por 255 várias vezes e ainda manter a variável original intacta.
Agradeço muito pela sua contribuição, com certeza poderá ajudar outros alunos!
Abraços e bons estudos!