Spaces:
Sleeping
Sleeping
File size: 1,950 Bytes
8f72b1f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
import argparse
def get_argparser():
parser = argparse.ArgumentParser("LOCA parser", add_help=False)
parser.add_argument('--model_name', default='loca_few_shot', type=str)
parser.add_argument(
'--data_path',
default='./data/FSC147_384_V2',
type=str
)
parser.add_argument(
'--model_path',
default='ckpt',
type=str
)
parser.add_argument('--backbone', default='resnet50', type=str)
parser.add_argument('--swav_backbone', action='store_true', default=True)
parser.add_argument('--reduction', default=8, type=int)
parser.add_argument('--image_size', default=512, type=int)
parser.add_argument('--num_enc_layers', default=3, type=int)
parser.add_argument('--num_ope_iterative_steps', default=3, type=int)
parser.add_argument('--emb_dim', default=256, type=int)
parser.add_argument('--num_heads', default=8, type=int)
parser.add_argument('--kernel_dim', default=3, type=int)
parser.add_argument('--num_objects', default=3, type=int)
parser.add_argument('--epochs', default=200, type=int)
parser.add_argument('--resume_training', action='store_true')
parser.add_argument('--lr', default=1e-4, type=float)
parser.add_argument('--backbone_lr', default=0, type=float)
parser.add_argument('--lr_drop', default=200, type=int)
parser.add_argument('--weight_decay', default=1e-4, type=float)
parser.add_argument('--batch_size', default=1, type=int)
parser.add_argument('--dropout', default=0.1, type=float)
parser.add_argument('--num_workers', default=8, type=int)
parser.add_argument('--max_grad_norm', default=0.1, type=float)
parser.add_argument('--aux_weight', default=0.3, type=float)
parser.add_argument('--tiling_p', default=0.5, type=float)
parser.add_argument('--zero_shot', action='store_true')
parser.add_argument('--pre_norm', action='store_true', default=True)
return parser
|