Spaces:
Runtime error
Runtime error
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) | |
])) | |