Olá, eu estava checando a questão da modificação das camadas para melhorar ou piorar a perda do modelo e notei uma coisa. Independente de restartar o runtime pra zerar as variáveis ocultas, o imagens_teste está sendo dividido por 255 a cada interação do modelo.
Isso porque o código imagens_teste = imagens_teste/float(255) está dentro do código do modelo, portanto cada iteração que iniciamos a variável é novamente dividida.
Eu não sei se isso influencia diretamente no modelo, mas creio que sim.
att,
Guilherme