()
| 422 | return |
| 423 | |
| 424 | def train(): |
| 425 | |
| 426 | print(parser.format_values()) |
| 427 | |
| 428 | # Load data |
| 429 | if args.dataset_type == '7Scenes': |
| 430 | |
| 431 | train_dl, val_dl, test_dl, hwf, i_split, near, far = load_7Scenes_dataloader(args) |
| 432 | near = near |
| 433 | far = far |
| 434 | print('NEAR FAR', near, far) |
| 435 | train_feature(args, train_dl, val_dl, test_dl, hwf, i_split, near, far) |
| 436 | return |
| 437 | |
| 438 | elif args.dataset_type == 'Cambridge': |
| 439 | |
| 440 | train_dl, val_dl, test_dl, hwf, i_split, near, far = load_Cambridge_dataloader(args) |
| 441 | near = near |
| 442 | far = far |
| 443 | |
| 444 | print('NEAR FAR', near, far) |
| 445 | train_feature(args, train_dl, val_dl, test_dl, hwf, i_split, near, far) |
| 446 | return |
| 447 | |
| 448 | else: |
| 449 | print('Unknown dataset type', args.dataset_type, 'exiting') |
| 450 | return |
| 451 | |
| 452 | if __name__ == "__main__": |
| 453 |
no test coverage detected