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