|
|
@ -216,6 +216,7 @@ class U2Trainer(Trainer):
|
|
|
|
msg += f"{v:>.8f}" if isinstance(v,
|
|
|
|
msg += f"{v:>.8f}" if isinstance(v,
|
|
|
|
float) else f"{v}"
|
|
|
|
float) else f"{v}"
|
|
|
|
msg += ","
|
|
|
|
msg += ","
|
|
|
|
|
|
|
|
msg = msg[:-1] # remove the last ","
|
|
|
|
if (batch_index + 1
|
|
|
|
if (batch_index + 1
|
|
|
|
) % self.config.training.log_interval == 0:
|
|
|
|
) % self.config.training.log_interval == 0:
|
|
|
|
logger.info(msg)
|
|
|
|
logger.info(msg)
|
|
|
|