StopIteration: Caught StopIteration in replica 0 on device 0

**Summary[问题简述（一句话）]**
0.0.1 版本代码按照训练训练合成器出错（机器上有多个gpu）

**Env & To Reproduce[复现与环境]**
描述你用的环境、代码版本、模型
环境：ubuntu 20.04
代码版本：0.0.1
python：3.9.6
torch.__version__: 2.0.0+cu117

**Screenshots[截图（如有）]**

No source files, tests, commands, or entry points are identified. Start by reproducing the synthesizer training failure on Ubuntu 20.04 with Python 3.9.6, PyTorch 2.0.0+cu117, version 0.0.1, and multiple GPUs; done means locating and resolving the StopIteration reported for replica 0.

Written by the indexing model from the issue text.