PFA-Demo / options.py
wissemkarous's picture
init
8c79f36 verified
raw
history blame
513 Bytes
gpu = "0"
random_seed = 0
data_type = "unseen"
video_path = "../lip/"
train_list = f"data/{data_type}_train.txt"
val_list = f"data/{data_type}_val.txt"
anno_path = "../GRID_align_txt"
vid_padding = 75
txt_padding = 200
batch_size = 8
base_lr = 2e-5
num_workers = 4
max_epoch = 90
display = 10
test_step = 1000
save_prefix = f"weights/LipNet_{data_type}"
is_optimize = True
device = "cpu"
two_stream_weights = "pretrain/LipNet_coords_loss_0.025581153109669685_wer_0.01746208431890914_cer_0.006488426950253695.pt"