Buenas Renzo,
por las dimensiones pareciera que estás teniendo problemas en la primer capa. No será que estás aplicando batch norm previo a la capa afin, en vez de posteriormente?
Saludos.
por las dimensiones pareciera que estás teniendo problemas en la primer capa. No será que estás aplicando batch norm previo a la capa afin, en vez de posteriormente?
Saludos.