repvgg_b3g4

Model info

Python source

#!/usr/bin/env python3
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
from basecls.configs import RepVGGConfig

_cfg = dict(
    model=dict(
        name="repvgg_b3g4",
    ),
    loss=dict(
        label_smooth=0.1,
    ),
    augments=dict(
        name="AutoAugment",
        mixup=dict(
            mixup_alpha=0.2,
            permute=1,
        ),
    ),
    solver=dict(
        max_epoch=200,
    ),
)


class Cfg(RepVGGConfig):
    def __init__(self, values_or_file=None, **kwargs):
        super().__init__(_cfg)
        self.merge(values_or_file, **kwargs)