Spaces:
Runtime error
Runtime error
| AMP: true | |
| CUDNN_BENCHMARK: true | |
| CUDNN_DETERMINISTIC: false | |
| DATA: | |
| EOS_INDEX: 2 | |
| IMAGE_CROP_SIZE: 224 | |
| IMAGE_TRANSFORM_TRAIN: | |
| - random_resized_crop | |
| - horizontal_flip | |
| - color_jitter | |
| - normalize | |
| IMAGE_TRANSFORM_VAL: | |
| - smallest_resize | |
| - center_crop | |
| - normalize | |
| MASKED_LM: | |
| MASK_PROBABILITY: 0.85 | |
| MASK_PROPORTION: 0.15 | |
| REPLACE_PROBABILITY: 0.1 | |
| MASK_INDEX: 3 | |
| MAX_CAPTION_LENGTH: 50 | |
| ROOT: datasets/redcaps/tarfiles/*.tar | |
| SOS_INDEX: 1 | |
| TOKENIZER_MODEL: datasets/common_30k.model | |
| UNK_INDEX: 0 | |
| USE_PERCENTAGE: 100.0 | |
| USE_SINGLE_CAPTION: false | |
| VOCAB_SIZE: 30000 | |
| MODEL: | |
| DECODER: | |
| BEAM_SIZE: 5 | |
| MAX_DECODING_STEPS: 30 | |
| NAME: nucleus_sampling | |
| NUCLEUS_SIZE: 0.9 | |
| LABEL_SMOOTHING: 0.1 | |
| NAME: virtex_web | |
| TEXTUAL: | |
| DROPOUT: 0.1 | |
| NAME: transdec_prenorm::L6_H512_A8_F2048 | |
| VISUAL: | |
| FEATURE_SIZE: 2048 | |
| FROZEN: false | |
| NAME: torchvision::resnet50 | |
| PRETRAINED: false | |
| OPTIM: | |
| BATCH_SIZE: 256 | |
| CLIP_GRAD_NORM: 10.0 | |
| CNN_LR: 0.0005 | |
| LOOKAHEAD: | |
| ALPHA: 0.5 | |
| STEPS: 5 | |
| USE: false | |
| LR: 0.0005 | |
| LR_DECAY_NAME: cosine | |
| LR_GAMMA: 0.1 | |
| LR_STEPS: [] | |
| NO_DECAY: .*textual.(embedding|transformer).*(norm.*|bias) | |
| NUM_ITERATIONS: 1500000 | |
| OPTIMIZER_NAME: adamw | |
| SGD_MOMENTUM: 0.9 | |
| WARMUP_STEPS: 10000 | |
| WEIGHT_DECAY: 0.01 | |
| RANDOM_SEED: 0 | |