Add files via upload

onnx的配置
pull/380/head
szsteven008 2024-12-31 10:55:51 +08:00 committed by GitHub
parent 66e8919f33
commit 7598ea558b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions

View File

@ -34,8 +34,8 @@ max_text_len = 256
text_encoder_type = "bert-base-uncased"
use_text_enhancer = True
use_fusion_layer = True
use_checkpoint = True
use_transformer_ckpt = True
use_checkpoint = False #True
use_transformer_ckpt = False #True
use_text_cross_attention = True
text_dropout = 0.0
fusion_dropout = 0.0

View File

@ -34,8 +34,8 @@ max_text_len = 256
text_encoder_type = "bert-base-uncased"
use_text_enhancer = True
use_fusion_layer = True
use_checkpoint = True
use_transformer_ckpt = True
use_checkpoint = False #True
use_transformer_ckpt = False #True
use_text_cross_attention = True
text_dropout = 0.0
fusion_dropout = 0.0