tzrec/main.py [446:462]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                    last_ckpt_step = i_step
                    checkpoint_util.save_model(
                        os.path.join(model_dir, f"model.ckpt-{i_step}"),
                        model,
                        optimizer,
                    )
                    if eval_dataloader is not None:
                        _evaluate(
                            model,
                            eval_dataloader,
                            eval_config,
                            eval_result_filename=eval_result_filename,
                            global_step=i_step,
                            eval_summary_writer=eval_summary_writer,
                            global_epoch=i_epoch,
                        )
                        model.train()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



tzrec/main.py [468:484]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                last_ckpt_step = i_step
                checkpoint_util.save_model(
                    os.path.join(model_dir, f"model.ckpt-{i_step}"),
                    model,
                    optimizer,
                )
                if eval_dataloader is not None:
                    _evaluate(
                        model,
                        eval_dataloader,
                        eval_config,
                        eval_result_filename=eval_result_filename,
                        global_step=i_step,
                        eval_summary_writer=eval_summary_writer,
                        global_epoch=i_epoch,
                    )
                    model.train()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -