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

Fix config of beit (#1528)

pull/1554/head
Yixiao Fang 2023-04-28 16:33:14 +08:00 committed by GitHub
parent b51d7d21de
commit a3fa328f09
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
configs/beit/beit_beit-base-p16_8xb256-amp-coslr-300e_in1k.py
View File

@ -57,6 +57,7 @@ model = dict(
patch_size=16,
drop_path_rate=0.1,
final_norm=True,
out_type='raw',
layer_scale_init_value=0.1,
init_cfg=[
dict(type='TruncNormal', std=0.02, layer='Linear'),

1
configs/beitv2/beitv2_beit-base-p16_8xb256-amp-coslr-1600e_in1k.py
View File

@ -37,6 +37,7 @@ model = dict(
out_indices=[-4, -1],
drop_path_rate=drop_path_rate,
final_norm=False,
out_type='raw',
layer_scale_init_value=layer_scale_init_value,
init_cfg=[
dict(type='TruncNormal', std=0.02, layer='Linear'),

1
configs/beitv2/beitv2_beit-base-p16_8xb256-amp-coslr-300e_in1k.py
View File

@ -37,6 +37,7 @@ model = dict(
out_indices=[-4, -1],
drop_path_rate=drop_path_rate,
final_norm=False,
out_type='raw',
layer_scale_init_value=layer_scale_init_value,
init_cfg=[
dict(type='TruncNormal', std=0.02, layer='Linear'),