提交 0adb8271 编写于 作者: 悟、's avatar 悟、 提交者: cuicheng01

fix cae dataset

上级 2a454abe
...@@ -17,7 +17,7 @@ Global: ...@@ -17,7 +17,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: cae_base_patch16_224 name: cae_base_patch16_224
class_num: 102 class_num: 1000
drop_rate: 0.0 drop_rate: 0.0
drop_path_rate: 0.1 drop_path_rate: 0.1
attn_drop_rate: 0.0 attn_drop_rate: 0.0
......
...@@ -17,7 +17,7 @@ Global: ...@@ -17,7 +17,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: cae_large_patch16_224 name: cae_large_patch16_224
class_num: 102 class_num: 1000
drop_rate: 0.0 drop_rate: 0.0
drop_path_rate: 0.2 drop_path_rate: 0.2
attn_drop_rate: 0.0 attn_drop_rate: 0.0
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册