Potential fix for pull request finding

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
Signed-off-by: Jinzhe Zeng <njzjz@qq.com>

Author: njzjz

deepmd/utils/batch_size.py

@@ -292,7 +292,12 @@ def is_oom_error(self, e: Exception) -> bool:
     def set_oom_retry_mode(self, enable: bool) -> None:
         """Set OOM retry mode.
 
-        In OOM retry mode, all data will be re-executed.
+        In OOM retry mode, an OOM during execution may reduce the current
+        batch size and raise :class:`RetrySignal` to indicate that execution
+        should be retried.
+
+        Callers that want all data to be re-executed must catch
+        :class:`RetrySignal` and restart the full evaluation themselves.
 
         Parameters
         ----------