mirrors/mmclassification
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 21:53:55 +08:00
Code Issues Projects Releases Wiki Activity

Fix ResNet rsb config

Browse Source
This commit is contained in:
mzr1996 2021-12-31 11:12:21 +08:00
parent e1a4020168
commit cdf569a805
2 changed files with 2 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
configs/resnet/resnet50_8xb256-rsb-a2-300e_in1k.py
View File

@ -11,12 +11,7 @@ model = dict(
norm_cfg=dict(type='SyncBN', requires_grad=True),
drop_path_rate=0.05,
),
head=dict(
loss=dict(
type='LabelSmoothLoss',
label_smooth_val=0.1,
mode='original',
)),
head=dict(loss=dict(use_sigmoid=True)),
train_cfg=dict(augments=[
dict(type='BatchMixup', alpha=0.1, num_classes=1000, prob=0.5),
dict(type='BatchCutMix', alpha=1.0, num_classes=1000, prob=0.5)

7
configs/resnet/resnet50_8xb256-rsb-a3-100e_in1k.py
View File

@ -8,12 +8,7 @@ _base_ = [
# Model settings
model = dict(
backbone=dict(norm_cfg=dict(type='SyncBN', requires_grad=True)),
head=dict(
loss=dict(
type='LabelSmoothLoss',
label_smooth_val=0.1,
mode='original',
)),
head=dict(loss=dict(use_sigmoid=True)),
train_cfg=dict(augments=[
dict(type='BatchMixup', alpha=0.1, num_classes=1000, prob=0.5),
dict(type='BatchCutMix', alpha=1.0, num_classes=1000, prob=0.5)