Fix bug that auto batch size doesn't consider distributed training#2533

eunwoosh merged 7 commits intoopen-edge-platform:releases/1.5.0open-edge-platform/training_extensions:releases/1.5.0from

eunwoosh:consider_dist_train_while_auto_bseunwoosh/training_extensions:consider_dist_train_while_auto_bsCopy head branch name to clipboard