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']