def __init__()

in pycls/models/efficientnet.py [0:0]

• 24 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self):
        assert cfg.TRAIN.DATASET in ['imagenet'], \
            'Training on {} is not supported'.format(cfg.TRAIN.DATASET)
        assert cfg.TEST.DATASET in ['imagenet'], \
            'Testing on {} is not supported'.format(cfg.TEST.DATASET)
        assert cfg.TRAIN.DATASET == cfg.TEST.DATASET, \
            'Train and test dataset must be the same for now'
        assert cfg.EFFICIENT_NET.HEAD_TYPE in ['conv_head', 'simple_head', 'linear_head'], \
            'Unsupported head type: {}'.format(cfg.EFFICIENT_NET.HEAD_TYPE)
        super(EfficientNet, self).__init__()
        self._construct_class(
            stem_w=cfg.EFFICIENT_NET.STEM_W,
            ds=cfg.EFFICIENT_NET.DEPTHS,
            ws=cfg.EFFICIENT_NET.WIDTHS,
            exp_rs=cfg.EFFICIENT_NET.EXP_RATIOS,
            se_r=cfg.EFFICIENT_NET.SE_RATIO,
            ss=cfg.EFFICIENT_NET.STRIDES,
            ks=cfg.EFFICIENT_NET.KERNELS,
            head_type=cfg.EFFICIENT_NET.HEAD_TYPE,
            head_w=cfg.EFFICIENT_NET.HEAD_W,
            act_type=cfg.EFFICIENT_NET.ACT_FUN,
            nc=cfg.MODEL.NUM_CLASSES
        )
        self.apply(nu.init_weights)