Haremos una distincin; si vamos a utilizar GPU para entrenar CPU, definiendo una variable llamada device. Ntese que tambin alteramos el tamao que usaremos de batch. En el caso de GPU, podemos utilizar valores 2 3 segn el tamao de memoria RAM que tenga la tarjeta grfica.