update some files

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

View File

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

View File

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

View File

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

View File

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