cd6ea74225b7c29238de13229aa1347db4098fea,train_model.py,,main,#,37

Before Change

            logger.info("model saved to: {:s}".format(path_model))
            logger.info("elapsed time: {:.1f} s".format(time.time() - time_start))
        df_losses = df_losses_curr
    logger.info("total training time: {:.1f} s".format(time.time() - time_start))

if __name__ == "__main__":
    main()

After Change

        if ((i + 1) % opts.iter_checkpoint == 0) or ((i + 1) == opts.n_iter):
            model.save_state(path_model)
            fnetlogger.to_csv(path_losses_csv)
            logger.info("loss log saved to: {:s}".format(path_losses_csv))
            logger.info("model saved to: {:s}".format(path_model))
            logger.info("elapsed time: {:.1f} s".format(time.time() - time_start))
    

Instances