From 2f51108beb6826242737cb21420af61fbee89ef6 Mon Sep 17 00:00:00 2001 From: Prajwal Date: Fri, 28 Aug 2020 23:27:06 +0530 Subject: [PATCH] Another undefined variable fix --- hq_wav2lip_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hq_wav2lip_train.py b/hq_wav2lip_train.py index 86db55c..1c56d52 100644 --- a/hq_wav2lip_train.py +++ b/hq_wav2lip_train.py @@ -193,7 +193,7 @@ for p in syncnet.parameters(): recon_loss = nn.L1Loss() def get_sync_loss(mel, g): g = g[:, :, :, g.size(3)//2:] - g = torch.cat([g[:, :, i] for i in range(hparams.syncnet_T)], dim=1) + g = torch.cat([g[:, :, i] for i in range(syncnet_T)], dim=1) # B, 3 * T, H//2, W a, v = syncnet(mel, g) y = torch.ones(g.size(0), 1).float().to(device)