Spaces:
No application file
No application file
File size: 697 Bytes
5aa312d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | #package overview
package_name: image_captioning
training_data_file: captions.txt
target: caption
pipeline_name: image_captioning
pipeline_save_file: image_captioning_output_v
features: image
# Alogrithm parameters
n_estimators: 100
max_depth: 10
ENCODER: "google/vit-base-patch16-224"
DECODER: "gpt2"
TRAIN_BATCH_SIZE: 8
VAL_BATCH_SIZE: 8
VAL_EPOCHS: 1
LR: 5e-5
SEED: 42
MAX_LEN: 128
SUMMARY_LEN: 20
WEIGHT_DECAY: 0.01
MEAN: (0.485, 0.456, 0.406)
STD: (0.229, 0.224, 0.225)
TRAIN_PCT: 0.95
NUM_WORKERS: 32 # Number of threads in the system
EPOCHS: 3
IMG_SIZE: 224
LABEL_MASK: -100
TOP_K: 1000
TOP_P: 0.95
EARLY_STOPPING: True
NGRAM_SIZE: 3
LEN_PENALTY: 2.0
NUM_BEAMS: 4
NUM_LOGGING_STEPS: 1024 |