update

examples/nx_clean_unet/step_2_train_model.py

@@ -242,8 +242,6 @@ def main():
             clean_audios = clean_audios.to(device)
             noisy_audios = noisy_audios.to(device)
             one_labels = torch.ones(clean_audios.shape[0]).to(device)
-            print(f"clean_audios: {clean_audios.shape}")
-            print(f"noisy_audios: {noisy_audios.shape}")
 
             audio_g = generator.forward(noisy_audios)
 

examples/nx_clean_unet/yaml/config.yaml

@@ -12,8 +12,8 @@ down_sampling_hidden_channels: 64
 down_sampling_kernel_size: 4
 down_sampling_stride: 2
 
-tsfm_hidden_size: 256
-tsfm_attention_heads: 4
+tsfm_hidden_size: 512
+tsfm_attention_heads: 8
 tsfm_num_blocks: 6
 tsfm_dropout_rate: 0.1