在 '__init__.py' 中找不到引用 'SummaryWriter'

`SummaryWriter` 是TensorBoard的一个模块，通常是在TensorFlow中使用的。如果你在使用PyTorch，你需要安装TensorBoardX库，它提供了与TensorBoard相同的功能。你可以使用以下命令安装：

pip install tensorboardX

然后，在你的代码中，用以下方式导入 `SummaryWriter`：

from tensorboardX import SummaryWriter

这样就可以使用 `SummaryWriter` 了。