espnet2.gan_codec.shared.quantizer.modules.distrib.broadcast_tensors

source

espnet2.gan_codec.shared.quantizer.modules.distrib.broadcast_tensors(tensors: Iterable[Tensor], src: int = 0)

Broadcast the tensors from the given parameters to all workers. This can be used to ensure that all workers have the same model to start with.