def dict_to_stats(cfg_dict):
set_cfg(cfg)
cfg_new = CN(cfg_dict)
cfg.merge_from_other_cfg(cfg_new)
stats = get_stats()
set_cfg(cfg)
return stats
def match_baseline_cfg(cfg_dict, cfg_dict_baseline, verbose=True):
'''
Match the computational budget of a given baseline model. THe current
configuration dictionary will be modifed and returned.
Args:
cfg_dict (dict): Current experiment's configuration
cfg_dict_baseline (dict): Baseline configuration
verbose (str, optional): If printing matched paramter conunts
'''
stats_baseline = dict_to_stats(cfg_dict_baseline)
set_cfg(cfg)
cfg_new = CN(cfg_dict)
cfg.merge_from_other_cfg(cfg_new)
stats = match_computation(stats_baseline, key=['gnn', 'dim_inner'])
if 'gnn' in cfg_dict:
cfg_dict['gnn']['dim_inner'] = cfg.gnn.dim_inner
else:
cfg_dict['gnn'] = {'dim_inner', cfg.gnn.dim_inner}
set_cfg(cfg)
if verbose:
print('Computational budget has matched: Baseline params {}, '
'Current params {}'.format(stats_baseline, stats))
return cfg_dict
def dict_to_stats(cfg_dict):
from yacs.config import CfgNode as CN
set_cfg(cfg)
cfg_new = CN(cfg_dict)
cfg.merge_from_other_cfg(cfg_new)
stats = get_stats()
set_cfg(cfg)
return stats