parent
66e8919f33
commit
7598ea558b
|
@ -34,8 +34,8 @@ max_text_len = 256
|
||||||
text_encoder_type = "bert-base-uncased"
|
text_encoder_type = "bert-base-uncased"
|
||||||
use_text_enhancer = True
|
use_text_enhancer = True
|
||||||
use_fusion_layer = True
|
use_fusion_layer = True
|
||||||
use_checkpoint = True
|
use_checkpoint = False #True
|
||||||
use_transformer_ckpt = True
|
use_transformer_ckpt = False #True
|
||||||
use_text_cross_attention = True
|
use_text_cross_attention = True
|
||||||
text_dropout = 0.0
|
text_dropout = 0.0
|
||||||
fusion_dropout = 0.0
|
fusion_dropout = 0.0
|
||||||
|
|
|
@ -34,8 +34,8 @@ max_text_len = 256
|
||||||
text_encoder_type = "bert-base-uncased"
|
text_encoder_type = "bert-base-uncased"
|
||||||
use_text_enhancer = True
|
use_text_enhancer = True
|
||||||
use_fusion_layer = True
|
use_fusion_layer = True
|
||||||
use_checkpoint = True
|
use_checkpoint = False #True
|
||||||
use_transformer_ckpt = True
|
use_transformer_ckpt = False #True
|
||||||
use_text_cross_attention = True
|
use_text_cross_attention = True
|
||||||
text_dropout = 0.0
|
text_dropout = 0.0
|
||||||
fusion_dropout = 0.0
|
fusion_dropout = 0.0
|
||||||
|
|
Loading…
Reference in New Issue