blob: fff73595a27233e3b9e4b629427dc949a31dea5b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
import os
import time
import json
from bunch import Bunch
def get_config_from_json(json_file):
with open(json_file, 'r') as config_file:
config_dict = json.load(config_file)
config = Bunch(config_dict)
return config
def process_config(json_file):
config = get_config_from_json(json_file)
config.config_file = json_file
config.exp_dir = os.path.join(
'experiments', time.strftime('%Y-%m-%d/', time.localtime()),
config.exp_name)
config.tensorboard_log_dir = os.path.join(
'experiments', time.strftime('%Y-%m-%d/', time.localtime()),
config.exp_name, 'logs/')
config.checkpoint_dir = os.path.join(
'experiments', time.strftime('%Y-%m-%d/', time.localtime()),
config.exp_name, 'checkpoints/')
return config
|
