mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 18:24:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

update some files

Browse Source
This commit is contained in:
Mingshuang Luo 2021-12-22 16:11:35 +08:00
parent a9abcc5fda
commit 1abf255bdd
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/grid/AVSR/lipnet_ctc_vsr/decode.py
View File

@ -142,7 +142,7 @@ def get_params() -> AttributeDict:
"anno_path": Path("download/GRID/GRID_align_txt"), "anno_path": Path("download/GRID/GRID_align_txt"),
"val_list": Path("download/GRID/unseen_val.txt"), "val_list": Path("download/GRID/unseen_val.txt"),
"vid_padding": 75, "vid_padding": 75,
"num_workers": 1, "num_workers": 16,
"batch_size": 120, "batch_size": 120,
} }
) )

2
egs/grid/AVSR/lipnet_ctc_vsr/train.py
View File

@ -181,7 +181,7 @@ def get_params() -> AttributeDict:
"train_list": Path("download/GRID/unseen_train.txt"), "train_list": Path("download/GRID/unseen_train.txt"),
"vid_padding": 75, "vid_padding": 75,
"aud_padding": 200, "aud_padding": 200,
"num_workers": 1, "num_workers": 16,
"batch_size": 120, "batch_size": 120,
} }
) )

2
egs/grid/AVSR/local/dataset_av.py
View File

@ -8,7 +8,7 @@ import torch
import torchaudio import torchaudio
from torch.utils.data import Dataset from torch.utils.data import Dataset
from cvtransforms import HorizontalFlip, ColorNormalize from .cvtransforms import HorizontalFlip, ColorNormalize
class dataset_av(Dataset): class dataset_av(Dataset):

2
egs/grid/AVSR/local/dataset_visual.py
View File

@ -4,7 +4,7 @@ import os
import numpy as np import numpy as np
import torch import torch
from torch.utils.data import Dataset from torch.utils.data import Dataset
from cvtransforms import HorizontalFlip, ColorNormalize from .cvtransforms import HorizontalFlip, ColorNormalize
class dataset_visual(Dataset): class dataset_visual(Dataset):