KyanChen's picture
init
f549064
raw
history blame
827 Bytes
model = dict(
type='ImageClassifier',
backbone=dict(
type='BEiT',
arch='eva-g',
img_size=224,
patch_size=14,
avg_token=True,
layer_scale_init_value=0.0,
output_cls_token=False,
use_abs_pos_emb=True,
use_rel_pos_bias=False,
use_shared_rel_pos_bias=False,
),
neck=None,
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=1408,
loss=dict(
type='LabelSmoothLoss', label_smooth_val=0.1, mode='original'),
),
init_cfg=[
dict(type='TruncNormal', layer='Linear', std=.02),
dict(type='Constant', layer='LayerNorm', val=1., bias=0.),
],
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0)
]))