| import argparse | |
| def get_args_parser(): | |
| parser = argparse.ArgumentParser('Holistic edge attention transformer', add_help=False) | |
| parser.add_argument('--exp_dataset', default='outdoor', | |
| help='the dataset for experiments, outdoor/s3d_floorplan') | |
| parser.add_argument('--lr', default=2e-4, type=float) | |
| parser.add_argument('--batch_size', default=16, type=int) | |
| parser.add_argument('--weight_decay', default=1e-5, type=float) | |
| parser.add_argument('--epochs', default=800, type=int) | |
| parser.add_argument('--lr_drop', default=600, type=int) | |
| parser.add_argument('--clip_max_norm', default=0.1, type=float, | |
| help='gradient clipping max norm') | |
| parser.add_argument('--print_freq', default=40, type=int) | |
| parser.add_argument('--output_dir', default='./checkpoints/ckpts_heat_outdoor_256', | |
| help='path where to save, empty for no saving') | |
| parser.add_argument('--resume', default='', | |
| help='resume from checkpoint') | |
| parser.add_argument('--start_epoch', default=0, type=int, metavar='N', | |
| help='start epoch') | |
| parser.add_argument('--num_workers', default=4, type=int) | |
| parser.add_argument('--image_size', default=256, type=int) | |
| parser.add_argument('--max_corner_num', default=150, type=int, | |
| help='the max number of corners allowed in the experiments') | |
| parser.add_argument('--corner_to_edge_multiplier', default=3, type=int, | |
| help='the max number of edges based on the number of corner candidates (assuming the ' | |
| 'average degree never greater than 6)') | |
| parser.add_argument('--lambda_corner', default=0.05, type=float, | |
| help='the max number of corners allowed in the experiments') | |
| parser.add_argument('--run_validation', action='store_true', | |
| help='Whether run validation or not, default: False') | |
| return parser | |