effnetv2_b0#
Model info#
Python source#
#!/usr/bin/env python3
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
from basecls.configs import EffNetConfig
_cfg = dict(
batch_size=64,
preprocess=dict(
img_size=192,
),
model=dict(
name="effnetv2_b0",
),
solver=dict(
basic_lr=0.1,
),
model_ema=dict(
enabled=True,
momentum=0.99996,
),
)
class Cfg(EffNetConfig):
def __init__(self, values_or_file=None, **kwargs):
super().__init__(_cfg)
self.merge(values_or_file, **kwargs)