Skip to content

head_configs

Bases: config

Source code in tinybig/config/head_configs.py 
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
class head_configs(config):
    def __init__(
        self,
        name: str = 'rpn_head',
        configs: dict = None,
        parameters: dict = None,
        m: int = None, n: int = None,
        batch: int = None,
        l: int = None,
        channel_num: int = 1,

        data_transformation_configs: dict = None,
        parameter_fabrication_configs: dict = None,
        remainder_configs: dict = None,

        channel_fusion_configs: dict = None,

        attribute_interdependence_configs: dict = None,
        instance_interdependence_configs: dict = None,

        input_processing_configs: dict | list = None,
        output_processing_configs: dict | list = None,

        device='cpu',
        *args, **kwargs
    ):
        if configs is not None:
            assert isinstance(configs, dict) and 'head_class' in configs and configs['head_class'] == 'tinybig.module.rpn_head'
            super().__init__(name=name, configs=configs, device=device, *args, **kwargs)
        elif parameters is not None:
            configs = {
                'head_class': 'tinybig.module.rpn_head',
                'head_parameters': parameters,
            }
            super().__init__(name=name, configs=configs, device=device, *args, **kwargs)
        elif m is not None and n is not None:
            parameters = {
                'name': name,
                'm': m,
                'n': n,
                'l': l,
                'batch': batch,
                'channel_num': channel_num,

                'data_transformation_configs': data_transformation_configs,
                'parameter_fabrication_configs': parameter_fabrication_configs,
                'remainder_configs': remainder_configs,
                'channel_fusion_configs': channel_fusion_configs,
                'attribute_interdependence_configs': attribute_interdependence_configs,
                'instance_interdependence_configs': instance_interdependence_configs,
                'input_processing_configs': input_processing_configs,
                'output_processing_configs': output_processing_configs,
                'device': device,
            }
            configs = {
                'head_class': 'tinybig.module.rpn_head',
                'head_parameters': parameters,
            }
            super().__init__(name=name, configs=configs, device=device, *args, **kwargs)
        else:
            super().__init__(name=name, configs=None, device=device, *args, **kwargs)

    def to_instantiation(self):
        if self.configs is None:
            raise ValueError('The head configs object cannot be none...')

        return config.instantiation_from_configs(
            configs=self.configs,
            device=self.device,
            class_name='head_class',
            parameter_name='head_parameters'
        )