i can't generate a checkpoint cause I get this error Exception compiling checkpoint: The size of ten
i can't generate a checkpoint cause I get this error Exception compiling checkpoint: The size of tensor a (4) must match the size of tensor b (320) at non-singleton dimension 1










