add use_fused_attn field
parent
46c1a0168c
commit
87ad22342e
|
@ -32,6 +32,8 @@ AMP:
|
|||
Arch:
|
||||
name: SwinTransformer_base_patch4_window12_384
|
||||
class_num: 1000
|
||||
# fused op can be used in AMP O2 mode only
|
||||
use_fused_attn: False
|
||||
|
||||
# loss function config for traing/eval process
|
||||
Loss:
|
||||
|
|
|
@ -33,6 +33,8 @@ Arch:
|
|||
name: SwinTransformer_base_patch4_window7_224
|
||||
class_num: 1000
|
||||
pretrained: True
|
||||
# fused op can be used in AMP O2 mode only
|
||||
use_fused_attn: False
|
||||
|
||||
# loss function config for traing/eval process
|
||||
Loss:
|
||||
|
|
|
@ -32,6 +32,8 @@ AMP:
|
|||
Arch:
|
||||
name: SwinTransformer_large_patch4_window12_384
|
||||
class_num: 1000
|
||||
# fused op can be used in AMP O2 mode only
|
||||
use_fused_attn: False
|
||||
|
||||
# loss function config for traing/eval process
|
||||
Loss:
|
||||
|
|
|
@ -32,6 +32,8 @@ AMP:
|
|||
Arch:
|
||||
name: SwinTransformer_large_patch4_window7_224
|
||||
class_num: 1000
|
||||
# fused op can be used in AMP O2 mode only
|
||||
use_fused_attn: False
|
||||
|
||||
# loss function config for traing/eval process
|
||||
Loss:
|
||||
|
|
|
@ -32,6 +32,8 @@ AMP:
|
|||
Arch:
|
||||
name: SwinTransformer_small_patch4_window7_224
|
||||
class_num: 1000
|
||||
# fused op can be used in AMP O2 mode only
|
||||
use_fused_attn: False
|
||||
|
||||
# loss function config for traing/eval process
|
||||
Loss:
|
||||
|
|
Loading…
Reference in New Issue