remove dependency on yaml
This commit is contained in:
parent
d14e6c5994
commit
72f01a2a2d
10
main.py
10
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']
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user