Andrej Karpathymore refactor35aa1102 years agoHistory
57 lines · 1.7 KB · C
RawDownload
/*Implements a simple logger that writes log files in the output directory.The Logger object is stateless and uses append mode to write to log files.*/#ifndef LOGGER_H#define LOGGER_H #include <assert.h>#include <string.h>// defines: fopenCheck, freadCheck, fcloseCheck, fseekCheck, mallocCheck#include "utils.h" typedef struct {    int active;    char output_log_file[512];} Logger; void logger_init(Logger *logger, const char *log_dir, int process_rank, int resume) {    // currently, only rank 0 writes logs    logger->active = 0;    if (log_dir != NULL && process_rank == 0) {        logger->active = 1;        assert(strlen(log_dir) < 500); // being a bit lazy, could relax later        snprintf(logger->output_log_file, 512, "%s/main.log", log_dir);        if (resume == 0) {            // wipe any existing logfile clean if we're starting fresh            FILE *logfile = fopenCheck(logger->output_log_file, "w");            fclose(logfile);        }    }} void logger_log_eval(Logger *logger, int step, float val) {    if (logger->active == 1) {        FILE *logfile = fopenCheck(logger->output_log_file, "a");        fprintf(logfile, "s:%d eval:%.4f\n", step, val);        fclose(logfile);    }} void logger_log_val(Logger *logger, int step, float val_loss) {    if (logger->active == 1) {        FILE *logfile = fopenCheck(logger->output_log_file, "a");        fprintf(logfile, "s:%d tel:%.4f\n", step, val_loss);        fclose(logfile);    }} void logger_log_train(Logger *logger, int step, float train_loss) {    if (logger->active == 1) {        FILE *logfile = fopenCheck(logger->output_log_file, "a");        fprintf(logfile, "s:%d trl:%.4f\n", step, train_loss);        fclose(logfile);    }} #endif