From 72f01a2a2d1b93c62c47d8c51773e57f1daf24e2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Christoffer=20M=C3=BCller=20Madsen?= Date: Sun, 20 May 2018 16:03:58 +0200 Subject: [PATCH] remove dependency on yaml --- main.py | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/main.py b/main.py index 53b0444..a22d236 100644 --- a/main.py +++ b/main.py @@ -77,15 +77,6 @@ if not os.path.isdir(model_path()): if not os.path.isdir(log_path): os.mkdir(log_path) - -def save_config(): - import yaml - # checkpoint_path = os.path.join(config['model_storage_path'], config['model']) - # config_path = os.path.join(checkpoint_path, 'config') - # with open(config_path, 'a+') as f: - # print("lol") - print(yaml.dump(config)) - # Define helper functions def log_train_outcome(outcome, diff_in_values, trained_eps = 0, log_path = os.path.join(model_path(), 'logs', "train.log")): format_vars = { 'trained_eps': trained_eps, @@ -172,7 +163,6 @@ if __name__ == "__main__": # Set up network from network import Network - save_config() # Set up variables episode_count = config['episode_count']