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
|