TODO: Docstring for log_train_step. Args: epoch (TYPE): NULL batch (TYPE): NULL steps_loss (TYPE): NULL cost_time (TYPE): NULL Returns: TODO Raises: NULL
(epoch, batch, steps_loss, cost_time)
| 37 | |
| 38 | |
| 39 | def log_train_step(epoch, batch, steps_loss, cost_time): |
| 40 | """TODO: Docstring for log_train_step. |
| 41 | |
| 42 | Args: |
| 43 | epoch (TYPE): NULL |
| 44 | batch (TYPE): NULL |
| 45 | steps_loss (TYPE): NULL |
| 46 | cost_time (TYPE): NULL |
| 47 | |
| 48 | Returns: TODO |
| 49 | |
| 50 | Raises: NULL |
| 51 | """ |
| 52 | if len(steps_loss) == 0: |
| 53 | return |
| 54 | |
| 55 | logging.info(f'[train] epoch {epoch}, batch {batch}. ' + \ |
| 56 | f'loss is {sum(steps_loss) / len(steps_loss):.10f}. ' + \ |
| 57 | f'cost {cost_time:.2f}s') |
| 58 | steps_loss.clear() |
| 59 | |
| 60 | |
| 61 | def epoch_train(config, model, optimizer, epoch, train_data, is_debug=False): |