File size: 3,138 Bytes
da6d0ff |
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 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
from .segmenter import CRIS
# from .segmenter_angular import CRIS_S
from .segmenter_verbonly import CRIS_PosOnly
from .segmenter_verbonly_fin import CRIS_PosOnly_rev
from .segmenter_verbonly_hardneg import CRIS_VerbOnly
from loguru import logger
def build_segmenter_pos_rev(args):
model = CRIS_PosOnly_rev(args)
backbone = []
head = []
for k, v in model.named_parameters():
if k.startswith('backbone') and 'positional_embedding' not in k:
backbone.append(v)
else:
head.append(v)
logger.info('Backbone with decay={}, Head={}'.format(len(backbone), len(head)))
param_list = [{
'params': backbone,
'initial_lr': args.lr_multi * args.base_lr
}, {
'params': head,
'initial_lr': args.base_lr
}]
return model, param_list
def build_segmenter_pos(args):
model = CRIS_PosOnly(args)
backbone = []
head = []
for k, v in model.named_parameters():
if k.startswith('backbone') and 'positional_embedding' not in k:
backbone.append(v)
else:
head.append(v)
logger.info('Backbone with decay={}, Head={}'.format(len(backbone), len(head)))
param_list = [{
'params': backbone,
'initial_lr': args.lr_multi * args.base_lr
}, {
'params': head,
'initial_lr': args.base_lr
}]
return model, param_list
def build_segmenter(args):
model = CRIS_VerbOnly(args)
backbone = []
head = []
for k, v in model.named_parameters():
if k.startswith('backbone') and 'positional_embedding' not in k:
backbone.append(v)
else:
head.append(v)
logger.info('Backbone with decay={}, Head={}'.format(len(backbone), len(head)))
param_list = [{
'params': backbone,
'initial_lr': args.lr_multi * args.base_lr
}, {
'params': head,
'initial_lr': args.base_lr
}]
return model, param_list
def build_segmenter_original(args):
model = CRIS(args)
backbone = []
head = []
for k, v in model.named_parameters():
if k.startswith('backbone') and 'positional_embedding' not in k:
backbone.append(v)
else:
head.append(v)
logger.info('Backbone with decay={}, Head={}'.format(len(backbone), len(head)))
param_list = [{
'params': backbone,
'initial_lr': args.lr_multi * args.base_lr
}, {
'params': head,
'initial_lr': args.base_lr
}]
return model, param_list
# def build_segmenter_textaug(args):
# model = CRIS_Wo_Noise(args)
# backbone = []
# head = []
# for k, v in model.named_parameters():
# if k.startswith('backbone') and 'positional_embedding' not in k:
# backbone.append(v)
# else:
# head.append(v)
# logger.info('Backbone with decay={}, Head={}'.format(len(backbone), len(head)))
# param_list = [{
# 'params': backbone,
# 'initial_lr': args.lr_multi * args.base_lr
# }, {
# 'params': head,
# 'initial_lr': args.base_lr
# }]
# return model, param_list
|