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))