|
|
@ -225,8 +225,8 @@ def run(rank, n_gpus, hps):
|
|
|
|
scaler = GradScaler(enabled=hps.train.fp16_run)
|
|
|
|
scaler = GradScaler(enabled=hps.train.fp16_run)
|
|
|
|
|
|
|
|
|
|
|
|
net_d=optim_d=scheduler_d=None
|
|
|
|
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):
|
|
|
|
for epoch in range(epoch_str, hps.train.epochs + 1):
|
|
|
|
print("start training from epoch %s"%epoch)
|
|
|
|
|
|
|
|
if rank == 0:
|
|
|
|
if rank == 0:
|
|
|
|
train_and_evaluate(
|
|
|
|
train_and_evaluate(
|
|
|
|
rank,
|
|
|
|
rank,
|
|
|
|