def init_post_backward_hook(self, bagua_module: BaguaModule): def hook(): if self._should_communicate(bagua_module): bagua_module.
確定! 回上一頁