mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix ResNet rsb config

pull/623/head
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)