update

examples/mpnet_aishell/step_2_train_model.py

@@ -171,7 +171,7 @@ def main():
     num_params = 0
     for p in generator.parameters():
         num_params += p.numel()
-    logger.info("Total Parameters (generator): {:.3f}M".format(num_params/1e6))
+    logger.info("total parameters (generator): {:.3f}M".format(num_params/1e6))
 
     optim_g = torch.optim.AdamW(generator.parameters(), config.learning_rate, betas=[config.adam_b1, config.adam_b2])
     optim_d = torch.optim.AdamW(discriminator.parameters(), config.learning_rate, betas=[config.adam_b1, config.adam_b2])
@@ -210,9 +210,9 @@ def main():
         )
         for batch in train_data_loader:
             clean_audio, noisy_audio = batch
-            clean_audio = torch.autograd.Variable(clean_audio.to(device, non_blocking=True))
-            noisy_audio = torch.autograd.Variable(noisy_audio.to(device, non_blocking=True))
-            one_labels = torch.ones(config.batch_size).to(device, non_blocking=True)
+            clean_audio = clean_audio.to(device)
+            noisy_audio = noisy_audio.to(device)
+            one_labels = torch.ones(config.batch_size).to(device)
 
             clean_mag, clean_pha, clean_com = mag_pha_stft(clean_audio, config.n_fft, config.hop_size, config.win_size, config.compress_factor)
             noisy_mag, noisy_pha, noisy_com = mag_pha_stft(noisy_audio, config.n_fft, config.hop_size, config.win_size, config.compress_factor)
@@ -293,8 +293,8 @@ def main():
         with torch.no_grad():
             for batch in valid_data_loader:
                 clean_audio, noisy_audio = batch
-                clean_audio = torch.autograd.Variable(clean_audio.to(device, non_blocking=True))
-                noisy_audio = torch.autograd.Variable(noisy_audio.to(device, non_blocking=True))
+                clean_audio = clean_audio.to(device)
+                noisy_audio = noisy_audio.to(device)
 
                 clean_mag, clean_pha, clean_com = mag_pha_stft(clean_audio, config.n_fft, config.hop_size, config.win_size, config.compress_factor)
                 noisy_mag, noisy_pha, noisy_com = mag_pha_stft(noisy_audio, config.n_fft, config.hop_size, config.win_size, config.compress_factor)