Tu as posté ça sur un colab? Ou un jupyter notebook?
C’est pas les réseaux que je maîtrise le mieux mais je peux regarder 
Tu peux aussi nous en dire plus sur le but final?
On peut pinger @blind_man et tu peux regarder son topic https://forum.geekzone.fr/t/le-deep-learning-pour-les-nuls/