search

bytedance / byteps

A high performance and generic framework for distributed DNN training
Other
3.62k stars 487 forks source link

tensorflow: fix bug in broadcast_variables #416

Closed pleasantrabbit closed 2 years ago

pleasantrabbit commented 2 years ago

When there's only one rank in total, broadcast_variables should still return a tf operation.

Signed-off-by: Yulu Jia yulu.jia@bytedance.com