From 81ef49f278d279eaa4399494ecb37f74c9aa3a92 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Wed, 23 Aug 2023 18:16:47 +0800
Subject: [PATCH] Update train_ms.py

---
 train_ms.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/train_ms.py b/train_ms.py
index ad57ef1..a2f6ab0 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -7,8 +7,8 @@ import torch
 from torch import nn, optim
 from torch.nn import functional as F
 from torch.utils.data import DataLoader
-from torch.utils.tensorboard import SummaryWriter
-#from tensorboardX import SummaryWriter
+#from torch.utils.tensorboard import SummaryWriter
+from wandb.tensorboard import WandbSummaryWriter as SummaryWriter
 import torch.multiprocessing as mp
 import torch.distributed as dist
 from torch.nn.parallel import DistributedDataParallel as DDP
@@ -190,7 +190,7 @@ def run(rank, n_gpus, hps):
         scheduler_g.step()
         scheduler_d.step()
         if net_dur_disc is not None:
-            cheduler_dur_disc.step()
+            scheduler_dur_disc.step()
 
 
 def train_and_evaluate(rank, epoch, hps, nets, optims, schedulers, scaler, loaders, logger, writers):