pingu_98/Wav2Lip
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Another undefined variable fix

Browse source
This commit is contained in:
Prajwal 2020-08-28 23:27:06 +05:30 committed by GitHub
parent 35d96b3281
commit 2f51108beb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
hq_wav2lip_train.py
View file

@ -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)