use has_grad instead of train_mode (#29309) (#29343)

* use has_grad instead of train_mode

* add vlog for debug

* fix ut

* fix ut