#4536 fix bug

Merged
chenyifan01 merged 2 commits from ai_task_refactor_v3_train into V20230718 9 months ago