Update train.py
This commit is contained in:
parent
05395d8bbe
commit
f190361191
2
train.py
2
train.py
|
@ -215,7 +215,7 @@ def train_and_evaluate(rank, epoch, hps, nets, optims, schedulers, scaler, loade
|
||||||
logger.info('Train Epoch: {} [{:.0f}%]'.format(
|
logger.info('Train Epoch: {} [{:.0f}%]'.format(
|
||||||
epoch,
|
epoch,
|
||||||
100. * batch_idx / len(train_loader)))
|
100. * batch_idx / len(train_loader)))
|
||||||
logger.info(f"Losses: {[x.item() for x in losses]}, step: {global_step}, lr: {lr}, reference_loss={reference_loss}")
|
logger.info(f"Losses: {[x.item() for x in losses]}, step: {global_step}, lr: {lr}, reference_loss: {reference_loss}")
|
||||||
|
|
||||||
scalar_dict = {"loss/g/total": loss_gen_all, "loss/d/total": loss_disc_all, "learning_rate": lr,
|
scalar_dict = {"loss/g/total": loss_gen_all, "loss/d/total": loss_disc_all, "learning_rate": lr,
|
||||||
"grad_norm_d": grad_norm_d, "grad_norm_g": grad_norm_g}
|
"grad_norm_d": grad_norm_d, "grad_norm_g": grad_norm_g}
|
||||||
|
|
Loading…
Reference in New Issue