if __name__ =='__main__':
parser = ArgumentParser()
parser.add_argument("--test_dir",type=str, default="test",help='Directory containing the test data (must have subdirectory noisy/)')
parser.add_argument("--enhanced_dir",type=str, default="test",help='Directory containing the enhanced data')
parser.add_argument("--ckpt",type=str, default="logs/epoch=326-step=408750.ckpt",help='Path to model checkpoint.')
parser.add_argument("--corrector",type=str, choices=("ald","langevin","none"), default="ald",help="Corrector class for the PC sampler.")
parser.add_argument("--corrector_steps",type=int, default=1,help="Number of corrector steps")
parser.add_argument("--snr",type=float, default=0.5,help="SNR value for (annealed) Langevin dynmaics.")
parser.add_argument("--N",type=int, default=30,help="Number of reverse steps")
args = parser.parse_args()