Add files via upload

This commit is contained in:
RVC-Boss 2025-02-23 20:26:18 +08:00 committed by GitHub
parent aa07216bba
commit fbb9f21e53
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 6 additions and 6 deletions

View File

@ -251,8 +251,8 @@ def run(rank, n_gpus, hps):
scaler = GradScaler(enabled=hps.train.fp16_run)
print("start training from epoch %s" % epoch_str)
for epoch in range(epoch_str, hps.train.epochs + 1):
print("start training from epoch %s"%epoch)
if rank == 0:
train_and_evaluate(
rank,
@ -282,7 +282,7 @@ def run(rank, n_gpus, hps):
)
scheduler_g.step()
scheduler_d.step()
print("training done")
print("training done")
def train_and_evaluate(

View File

@ -225,8 +225,8 @@ def run(rank, n_gpus, hps):
scaler = GradScaler(enabled=hps.train.fp16_run)
net_d=optim_d=scheduler_d=None
print("start training from epoch %s" % epoch_str)
for epoch in range(epoch_str, hps.train.epochs + 1):
print("start training from epoch %s"%epoch)
if rank == 0:
train_and_evaluate(
rank,
@ -256,7 +256,7 @@ def run(rank, n_gpus, hps):
)
scheduler_g.step()
# scheduler_d.step()
print("training done")
print("training done")
def train_and_evaluate(

View File

@ -198,8 +198,8 @@ def run(rank, n_gpus, hps):
scaler = GradScaler(enabled=hps.train.fp16_run)
net_d=optim_d=scheduler_d=None
print("start training from epoch %s"%epoch_str)
for epoch in range(epoch_str, hps.train.epochs + 1):
print("start training from epoch %s"%epoch)
if rank == 0:
train_and_evaluate(
rank,
@ -228,7 +228,7 @@ def run(rank, n_gpus, hps):
None,
)
scheduler_g.step()
print("training done")
print("training done")
def train_and_evaluate(
rank, epoch, hps, nets, optims, schedulers, scaler, loaders, logger, writers