From 35d96b32814cb9684e0af64b901f9f114f8a5f24 Mon Sep 17 00:00:00 2001 From: Prajwal Date: Fri, 28 Aug 2020 21:14:00 +0530 Subject: [PATCH] Reduce sync loss weight --- wav2lip_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wav2lip_train.py b/wav2lip_train.py index 60dbfcc..ebc1a3c 100644 --- a/wav2lip_train.py +++ b/wav2lip_train.py @@ -251,7 +251,7 @@ def train(device, model, train_data_loader, test_data_loader, optimizer, average_sync_loss = eval_model(test_data_loader, global_step, device, model, checkpoint_dir) if average_sync_loss < .75: - hparams.set_hparam('syncnet_wt', 0.03) + hparams.set_hparam('syncnet_wt', 0.01) # without image GAN a lesser weight is sufficient prog_bar.set_description('L1: {}, Sync Loss: {}'.format(running_l1_loss / (step + 1), running_sync_loss / (step + 1)))