Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
mmpretrain
Commits
495d9ed9
Commit
495d9ed9
authored
Jun 24, 2025
by
limm
Browse files
add part code
parent
59b09903
Pipeline
#2799
canceled with stages
Changes
1000
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
309 additions
and
0 deletions
+309
-0
configs/_base_/models/efficientnet_b4.py
configs/_base_/models/efficientnet_b4.py
+12
-0
configs/_base_/models/efficientnet_b5.py
configs/_base_/models/efficientnet_b5.py
+12
-0
configs/_base_/models/efficientnet_b6.py
configs/_base_/models/efficientnet_b6.py
+12
-0
configs/_base_/models/efficientnet_b7.py
configs/_base_/models/efficientnet_b7.py
+12
-0
configs/_base_/models/efficientnet_b8.py
configs/_base_/models/efficientnet_b8.py
+12
-0
configs/_base_/models/efficientnet_em.py
configs/_base_/models/efficientnet_em.py
+13
-0
configs/_base_/models/efficientnet_es.py
configs/_base_/models/efficientnet_es.py
+13
-0
configs/_base_/models/efficientnet_l2.py
configs/_base_/models/efficientnet_l2.py
+12
-0
configs/_base_/models/efficientnet_v2/efficientnetv2_b0.py
configs/_base_/models/efficientnet_v2/efficientnetv2_b0.py
+12
-0
configs/_base_/models/efficientnet_v2/efficientnetv2_b1.py
configs/_base_/models/efficientnet_v2/efficientnetv2_b1.py
+12
-0
configs/_base_/models/efficientnet_v2/efficientnetv2_b2.py
configs/_base_/models/efficientnet_v2/efficientnetv2_b2.py
+12
-0
configs/_base_/models/efficientnet_v2/efficientnetv2_b3.py
configs/_base_/models/efficientnet_v2/efficientnetv2_b3.py
+12
-0
configs/_base_/models/efficientnet_v2/efficientnetv2_l.py
configs/_base_/models/efficientnet_v2/efficientnetv2_l.py
+12
-0
configs/_base_/models/efficientnet_v2/efficientnetv2_m.py
configs/_base_/models/efficientnet_v2/efficientnetv2_m.py
+12
-0
configs/_base_/models/efficientnet_v2/efficientnetv2_s.py
configs/_base_/models/efficientnet_v2/efficientnetv2_s.py
+12
-0
configs/_base_/models/efficientnet_v2/efficientnetv2_xl.py
configs/_base_/models/efficientnet_v2/efficientnetv2_xl.py
+12
-0
configs/_base_/models/eva/eva-g.py
configs/_base_/models/eva/eva-g.py
+29
-0
configs/_base_/models/eva/eva-l.py
configs/_base_/models/eva/eva-l.py
+30
-0
configs/_base_/models/hivit/base_224.py
configs/_base_/models/hivit/base_224.py
+28
-0
configs/_base_/models/hivit/small_224.py
configs/_base_/models/hivit/small_224.py
+28
-0
No files found.
Too many changes to show.
To preserve performance only
1000 of 1000+
files are displayed.
Plain diff
Email patch
configs/_base_/models/efficientnet_b4.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNet'
,
arch
=
'b4'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1792
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_b5.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNet'
,
arch
=
'b5'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
2048
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_b6.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNet'
,
arch
=
'b6'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
2304
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_b7.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNet'
,
arch
=
'b7'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
2560
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_b8.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNet'
,
arch
=
'b8'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
2816
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_em.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
# `em` means EfficientNet-EdgeTPU-M arch
backbone
=
dict
(
type
=
'EfficientNet'
,
arch
=
'em'
,
act_cfg
=
dict
(
type
=
'ReLU'
)),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1280
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_es.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
# `es` means EfficientNet-EdgeTPU-S arch
backbone
=
dict
(
type
=
'EfficientNet'
,
arch
=
'es'
,
act_cfg
=
dict
(
type
=
'ReLU'
)),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1280
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_l2.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNet'
,
arch
=
'l2'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
5504
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_v2/efficientnetv2_b0.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNetV2'
,
arch
=
'b0'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1280
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_v2/efficientnetv2_b1.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNetV2'
,
arch
=
'b1'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1280
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_v2/efficientnetv2_b2.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNetV2'
,
arch
=
'b2'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1408
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_v2/efficientnetv2_b3.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNetV2'
,
arch
=
'b3'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1536
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_v2/efficientnetv2_l.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNetV2'
,
arch
=
'l'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1280
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_v2/efficientnetv2_m.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNetV2'
,
arch
=
'm'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1280
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_v2/efficientnetv2_s.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNetV2'
,
arch
=
's'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1280
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/efficientnet_v2/efficientnetv2_xl.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'EfficientNetV2'
,
arch
=
'xl'
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1280
,
loss
=
dict
(
type
=
'CrossEntropyLoss'
,
loss_weight
=
1.0
),
topk
=
(
1
,
5
),
))
configs/_base_/models/eva/eva-g.py
0 → 100644
View file @
495d9ed9
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'BEiTViT'
,
arch
=
'eva-g'
,
img_size
=
224
,
patch_size
=
14
,
layer_scale_init_value
=
0.0
,
out_type
=
'avg_featmap'
,
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
)
]))
configs/_base_/models/eva/eva-l.py
0 → 100644
View file @
495d9ed9
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'BEiTViT'
,
arch
=
'l'
,
img_size
=
224
,
patch_size
=
14
,
layer_scale_init_value
=
0.0
,
out_type
=
'avg_featmap'
,
use_abs_pos_emb
=
True
,
use_rel_pos_bias
=
False
,
use_shared_rel_pos_bias
=
False
,
layer_cfgs
=
dict
(
bias
=
True
),
),
neck
=
None
,
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
1024
,
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
)
]))
configs/_base_/models/hivit/base_224.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'HiViT'
,
arch
=
'base'
,
img_size
=
224
,
ape
=
True
,
rpe
=
True
,
drop_path_rate
=
0.5
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
512
,
init_cfg
=
None
,
# suppress the default init_cfg of LinearClsHead.
loss
=
dict
(
type
=
'LabelSmoothLoss'
,
label_smooth_val
=
0.1
,
mode
=
'original'
),
cal_acc
=
False
),
init_cfg
=
[
dict
(
type
=
'TruncNormal'
,
layer
=
'Linear'
,
std
=
0.02
,
bias
=
0.
),
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
)
]),
)
configs/_base_/models/hivit/small_224.py
0 → 100644
View file @
495d9ed9
# model settings
model
=
dict
(
type
=
'ImageClassifier'
,
backbone
=
dict
(
type
=
'HiViT'
,
arch
=
'small'
,
img_size
=
224
,
ape
=
True
,
rpe
=
True
,
drop_path_rate
=
0.3
),
neck
=
dict
(
type
=
'GlobalAveragePooling'
),
head
=
dict
(
type
=
'LinearClsHead'
,
num_classes
=
1000
,
in_channels
=
384
,
init_cfg
=
None
,
# suppress the default init_cfg of LinearClsHead.
loss
=
dict
(
type
=
'LabelSmoothLoss'
,
label_smooth_val
=
0.1
,
mode
=
'original'
),
cal_acc
=
False
),
init_cfg
=
[
dict
(
type
=
'TruncNormal'
,
layer
=
'Linear'
,
std
=
0.02
,
bias
=
0.
),
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
)
]),
)
Prev
1
…
3
4
5
6
7
8
9
10
11
…
50
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment