File size: 1,911 Bytes
2d06dcc | 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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 | import argparse
import sys
import os
import importlib
from easydict import EasyDict
class ParamManager:
def __init__(self, args):
output_path_param = self.add_output_path_param(args)
method_param = self.get_method_param(args)
self.args = EasyDict(
dict(
vars(args),
**output_path_param,
**method_param
)
)
def get_method_param(self, args):
if args.config_file_name.endswith('.py'):
module_name = '.' + args.config_file_name[:-3]
else:
module_name = '.' + args.config_file_name
config = importlib.import_module(module_name, 'configs')
method_param = config.Param
method_args = method_param(args)
return method_args.hyper_param
def add_output_path_param(self, args):
task_output_dir = os.path.join(args.output_dir, args.type)
if not os.path.exists(task_output_dir):
os.makedirs(task_output_dir)
concat_names = [args.method, args.dataset, args.known_cls_ratio, args.labeled_ratio, args.backbone, args.seed]
method_output_name = "_".join([str(x) for x in concat_names])
method_output_dir = os.path.join(task_output_dir, method_output_name)
if not os.path.exists(method_output_dir):
os.makedirs(method_output_dir)
model_output_dir = os.path.join(method_output_dir, args.model_dir)
if not os.path.exists(model_output_dir):
os.makedirs(model_output_dir)
output_path_param = {
'method_output_dir': method_output_dir,
'model_output_dir': model_output_dir
}
return output_path_param |