()
| 5 | device = torch.device(f"cuda:{0}") |
| 6 | |
| 7 | def parse_args(): |
| 8 | parser = argparse.ArgumentParser() |
| 9 | parser.add_argument('--pred_wavsdir',type=str) |
| 10 | parser.add_argument('--gt_wavsdir',type=str) |
| 11 | args = parser.parse_args() |
| 12 | return args |
| 13 | |
| 14 | if __name__ == '__main__': |
| 15 | args = parse_args() |