(args)
| 1 | import os |
| 2 | |
| 3 | def parse_resultsdir(args): |
| 4 | if args.expname is None: |
| 5 | rsdir = os.path.join(args.results_dir, args.dataset) |
| 6 | else: |
| 7 | rsdir = os.path.join(args.results_dir, args.dataset, args.expname) |
| 8 | |
| 9 | os.makedirs(rsdir, exist_ok=True) |
| 10 | return rsdir |
| 11 | |
| 12 | |
| 13 | def find_best_checkpoint(directory, pattern, verbose=False): |