예제 #1
0
    def __init__(self, log_dir=None, hps={}, save_every=1):
        SummaryWriter.__init__(self, log_dir=log_dir)
        Logger.__init__(self, **hps)
        self.save_every = save_every
        f = open(log_dir + "/params.json", "wt")
        f.write(str(hps) + "\n")
        f.close()
        self.log_dir = log_dir
        outfile = open(log_dir + "/params.pickle", 'wb')
        pickle.dump(hps, outfile)
        outfile.close()

        self.last_csv_update_iteration = 0
        self.csvname = log_dir + "/db.csv"
        self.add_text("Hyperparameters", str(hps))