SLT-space / configs /predict_config_demo.yaml
plice13's picture
final demo #1
988f575 verified
Raw History Blame
1.29 kB
ModelArguments:
base_model_name: google/t5-v1_1-base
hidden_dropout_prob: 0.1
num_beams: 5
max_length: 128
top_k: 50
top_p: 0.9
temperature: 1.0
length_penalty: 1.0
repetition_penalty: 1.0
early_stopping: False
no_repeat_ngram_size: 3
do_sample: False
EvaluationArguments:
output_dir: ./results
model_name: T5_YTASL_pretrain
skip_frames: False
# Data processing
split: test
max_sequence_length: 250
max_token_length: 128
# Generation parameters
model_dir: ./checkpoints/t5-v1_1-base/model.safetensors
batch_size: 1
# Debugging
max_val_samples: none
SignDataArguments:
data_dir: ./data
annotation_path:
train: YT.annotations.train.json
dev: YT.annotations.dev.json
test: YT.annotations.dev.json
visual_features:
sign2vec:
enable_input: False
test: sign2vec/metadata_sign2vec.dev.json
mae:
enable_input: False
test: mae/metadata_mae.dev.json
dino:
enable_input: False
test: dino/metadata_dino.dev.json
pose:
enable_input: True
test: YouTubeASL.keypoints.dev.json
SignModelArguments:
projectors:
sign2vec:
dim: 768
mae:
dim: 768
dino:
dim: 1152
pose:
dim: 208