コード例 #1
0
ファイル: config_utils.py プロジェクト: MorganeAyle/SNIP-it
def autoconfig(config):
    print("setting autoconfig")
    temp_loader = DataManager(os.path.join(".", "utils"))
    auto_configuration = temp_loader.load_json("autoconfig")
    if config['data_set'] in auto_configuration["dataset"]:
        for key, value in auto_configuration["dataset"][
                config['data_set']].items():
            setattr(config, key, value)
    if config['model'] in auto_configuration["model"]:
        for key, value in auto_configuration["model"][config['model']].items():
            setattr(config, key, value)
    if config['l0']:
        for key, value in auto_configuration["l0"].items():
            # setattr(config, key, value)
            config['bla'] = value
            print(config[key])
    if config['prune_criterion'] in auto_configuration:
        for key, value in auto_configuration[
                config['prune_criterion']].items():
            setattr(config, key, value)
    if config['hoyer_square']:
        for key, value in auto_configuration["hoyer_square"].items():
            setattr(config, key, value)
    if config['group_hoyer_square']:
        for key, value in auto_configuration["group_hoyer_square"].items():
            setattr(config, key, value)
    return config
コード例 #2
0
def autoconfig(config):
    print("setting autoconfig")
    temp_loader = DataManager(os.path.join(".", "utils"))
    auto_configuration = temp_loader.load_json("autoconfig")
    if config.data_set in auto_configuration["dataset"]:
        for key, value in auto_configuration["dataset"][config.data_set].items():
            setattr(config, key, value)
    if config.model in auto_configuration["model"]:
        for key, value in auto_configuration["model"][config.model].items():
            setattr(config, key, value)
    if config.l0:
        for key, value in auto_configuration["l0"].items():
            setattr(config, key, value)
    if config.prune_criterion in auto_configuration:
        for key, value in auto_configuration[config.prune_criterion].items():
            setattr(config, key, value)
    if config.hoyer_square:
        for key, value in auto_configuration["hoyer_square"].items():
            setattr(config, key, value)
    if config.group_hoyer_square:
        for key, value in auto_configuration["group_hoyer_square"].items():
            setattr(config, key, value)