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

update lipnet-ctc-vsr for grid

Browse Source
This commit is contained in:
Mingshuang Luo 2021-12-24 14:16:12 +08:00
parent f033379805
commit 149ccd1b85
2 changed files with 2 additions and 2 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

@ -440,7 +440,7 @@ def main():
else: else:
G = None G = None
model = LipNet(num_classes=max_token_id+1) model = LipNet(num_classes=max_token_id + 1)
if params.avg == 1: if params.avg == 1:
load_checkpoint(f"{params.exp_dir}/epoch-{params.epoch}.pt", model) load_checkpoint(f"{params.exp_dir}/epoch-{params.epoch}.pt", model)
else: else:

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

@ -509,7 +509,7 @@ def run(rank, world_size, args):
device = torch.device("cuda", rank) device = torch.device("cuda", rank)
graph_compiler = CtcTrainingGraphCompiler(lexicon=lexicon, device=device) graph_compiler = CtcTrainingGraphCompiler(lexicon=lexicon, device=device)
model = LipNet(num_classes=max_token_id+1) model = LipNet(num_classes=max_token_id + 1)
checkpoints = load_checkpoint_if_available(params=params, model=model) checkpoints = load_checkpoint_if_available(params=params, model=model)