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()
🌈write in front :🔍个人主页 : 啊森要自信的主页 🌈作者寄语 🌈: 小菜鸟的力量不在于它的体型,而在于它内心的勇气和无限的潜能,只要你有决心,就没有什么事情是不可能的…