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)