"mace-bench/3rdparty/vscode:/vscode.git/clone" did not exist on "fb246ae0dcdfbfe5cf2360466ecc7d9b1e38c2b5"
Commit 495d9ed9 authored by limm's avatar limm
Browse files

add part code

parent 59b09903
Pipeline #2799 canceled with stages
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(type='ConvMixer', arch='1024/20'),
neck=dict(type='GlobalAveragePooling'),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=1024,
loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
))
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(type='ConvMixer', arch='1536/20'),
neck=dict(type='GlobalAveragePooling'),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=1536,
loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
))
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(type='ConvMixer', arch='768/32', act_cfg=dict(type='ReLU')),
neck=dict(type='GlobalAveragePooling'),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=768,
loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
))
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(type='ConvNeXt', arch='base', drop_path_rate=0.5),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=1024,
loss=dict(
type='LabelSmoothLoss', label_smooth_val=0.1, mode='original'),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0),
]),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(type='ConvNeXt', arch='large', drop_path_rate=0.5),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=1536,
loss=dict(
type='LabelSmoothLoss', label_smooth_val=0.1, mode='original'),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0),
]),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(type='ConvNeXt', arch='small', drop_path_rate=0.4),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=768,
loss=dict(
type='LabelSmoothLoss', label_smooth_val=0.1, mode='original'),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0),
]),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(type='ConvNeXt', arch='tiny', drop_path_rate=0.1),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=768,
loss=dict(
type='LabelSmoothLoss', label_smooth_val=0.1, mode='original'),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0),
]),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(type='ConvNeXt', arch='xlarge', drop_path_rate=0.5),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=2048,
loss=dict(
type='LabelSmoothLoss', label_smooth_val=0.1, mode='original'),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0),
]),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(
type='ConvNeXt',
arch='atto',
drop_path_rate=0.1,
layer_scale_init_value=0.,
use_grn=True,
),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=320,
loss=dict(type='LabelSmoothLoss', label_smooth_val=0.2),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(
type='ConvNeXt',
arch='base',
drop_path_rate=0.1,
layer_scale_init_value=0.,
use_grn=True,
),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=1024,
loss=dict(type='LabelSmoothLoss', label_smooth_val=0.1),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0),
]),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(
type='ConvNeXt',
arch='femto',
drop_path_rate=0.1,
layer_scale_init_value=0.,
use_grn=True,
),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=384,
loss=dict(type='LabelSmoothLoss', label_smooth_val=0.1),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(
type='ConvNeXt',
arch='huge',
drop_path_rate=0.1,
layer_scale_init_value=0.,
use_grn=True,
),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=2816,
loss=dict(type='LabelSmoothLoss', label_smooth_val=0.1),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0),
]),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(
type='ConvNeXt',
arch='large',
drop_path_rate=0.1,
layer_scale_init_value=0.,
use_grn=True,
),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=1536,
loss=dict(type='LabelSmoothLoss', label_smooth_val=0.1),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0),
]),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(
type='ConvNeXt',
arch='nano',
drop_path_rate=0.1,
layer_scale_init_value=0.,
use_grn=True,
),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=640,
loss=dict(type='LabelSmoothLoss', label_smooth_val=0.2),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(
type='ConvNeXt',
arch='pico',
drop_path_rate=0.1,
layer_scale_init_value=0.,
use_grn=True,
),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=512,
loss=dict(type='LabelSmoothLoss', label_smooth_val=0.1),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
)
# Model settings
model = dict(
type='ImageClassifier',
backbone=dict(
type='ConvNeXt',
arch='tiny',
drop_path_rate=0.2,
layer_scale_init_value=0.,
use_grn=True,
),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=768,
loss=dict(type='LabelSmoothLoss', label_smooth_val=0.2),
init_cfg=None,
),
init_cfg=dict(
type='TruncNormal', layer=['Conv2d', 'Linear'], std=.02, bias=0.),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0),
]),
)
model = dict(
type='ImageClassifier',
backbone=dict(
type='DaViT', arch='base', out_indices=(3, ), drop_path_rate=0.4),
neck=dict(type='GlobalAveragePooling'),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=1024,
loss=dict(
type='LabelSmoothLoss', label_smooth_val=0.1, mode='original'),
),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0)
]))
model = dict(
type='ImageClassifier',
backbone=dict(
type='DaViT', arch='small', out_indices=(3, ), drop_path_rate=0.2),
neck=dict(type='GlobalAveragePooling'),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=768,
loss=dict(
type='LabelSmoothLoss', label_smooth_val=0.1, mode='original'),
),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0)
]))
model = dict(
type='ImageClassifier',
backbone=dict(
type='DaViT', arch='t', out_indices=(3, ), drop_path_rate=0.1),
neck=dict(type='GlobalAveragePooling'),
head=dict(
type='LinearClsHead',
num_classes=1000,
in_channels=768,
loss=dict(
type='LabelSmoothLoss', label_smooth_val=0.1, mode='original'),
),
train_cfg=dict(augments=[
dict(type='Mixup', alpha=0.8),
dict(type='CutMix', alpha=1.0)
]))
model = dict(
type='ImageClassifier',
backbone=dict(
type='DeiT3',
arch='b',
img_size=224,
patch_size=16,
drop_path_rate=0.2),
neck=None,
head=dict(
type='VisionTransformerClsHead',
num_classes=1000,
in_channels=768,
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)
]))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment