update

examples/clean_unet_aishell/step_2_train_model.py

@@ -34,6 +34,8 @@ from toolbox.torchaudio.models.clean_unet.training import LinearWarmupCosineDeca
 from toolbox.torchaudio.models.clean_unet.loss import MultiResolutionSTFTLoss
 from toolbox.torchaudio.models.clean_unet.metrics import batch_pesq
 
+torch.autograd.set_detect_anomaly(True)
+
 
 def get_args():
     parser = argparse.ArgumentParser()