Skip to content
Snippets Groups Projects
Commit c8bfd29b authored by Bobholamovic's avatar Bobholamovic
Browse files

Fix load yaml

parent 15422bd7
No related branches found
No related tags found
No related merge requests found
...@@ -4,7 +4,7 @@ import os ...@@ -4,7 +4,7 @@ import os
import shutil import shutil
import random import random
import ast import ast
from os.path import basename, exists from os.path import basename, exists, splitext
import torch import torch
import torch.backends.cudnn as cudnn import torch.backends.cudnn as cudnn
...@@ -16,15 +16,14 @@ from utils.misc import OutPathGetter, Logger, register ...@@ -16,15 +16,14 @@ from utils.misc import OutPathGetter, Logger, register
def read_config(config_path): def read_config(config_path):
f = open(config_path, 'r') with open(config_path, 'r') as f:
cfg = yaml.load(f.read(), Loader=yaml.FullLoader) cfg = yaml.load(f.read(), Loader=yaml.FullLoader)
f.close()
return cfg or {} return cfg or {}
def parse_config(cfg_name, cfg): def parse_config(cfg_name, cfg):
# Parse the name of config file # Parse the name of config file
sp = cfg_name.split('.')[0].split('_') sp = splitext(cfg_name)[0].split('_')
if len(sp) >= 2: if len(sp) >= 2:
cfg.setdefault('tag', sp[1]) cfg.setdefault('tag', sp[1])
cfg.setdefault('suffix', '_'.join(sp[2:])) cfg.setdefault('suffix', '_'.join(sp[2:]))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment