Spaces:
Running
on
T4
Running
on
T4
Changed checkpointing in config
Browse files- cfg_app.py +2 -2
cfg_app.py
CHANGED
@@ -38,8 +38,8 @@ max_text_len = 256
|
|
38 |
text_encoder_type = "bert-base-uncased"
|
39 |
use_text_enhancer = True
|
40 |
use_fusion_layer = True
|
41 |
-
use_checkpoint =
|
42 |
-
use_transformer_ckpt =
|
43 |
use_text_cross_attention = True
|
44 |
text_dropout = 0.0
|
45 |
fusion_dropout = 0.0
|
|
|
38 |
text_encoder_type = "bert-base-uncased"
|
39 |
use_text_enhancer = True
|
40 |
use_fusion_layer = True
|
41 |
+
use_checkpoint = False
|
42 |
+
use_transformer_ckpt = False
|
43 |
use_text_cross_attention = True
|
44 |
text_dropout = 0.0
|
45 |
fusion_dropout = 0.0
|