Spaces:
Sleeping
Sleeping
File size: 513 Bytes
8c79f36 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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"
|