[BUG] with gradm and learning mode (2.0.2)