Fixed reporting average update in CPU backend when momentum iz zero