def inst_mt_res18_faster(args):
args = basic_setting(args)
args = basic_mean_teacher(args)
args = bs_128_setting(args)
args = basic_inst_setting(args)
args = mt_fst_ramp_down_setting(args)
args.network_func = 'get_mean_teacher_and_inst_resnet_18'
return args
def res18_cmb_rp_inst(args):
args = shared_sts.basic_setting(args)
args = save_setting(args)
args = shared_sts.bs_64_setting(args)
args = cmb_rp_inst_setting(args)
args.expId = 'cmb_rp_inst'
args.lr_boundaries = '2500011'
return args
def mt_part3_res18(args):
args = basic_setting(args)
args = basic_mean_teacher(args)
args = bs_128_setting(args)
args = mt_fst_ramp_down_setting(args)
args.whichimagenet = 'part3'
args.expId = 'mt_part3_res18'
return args
def mt_part10_res18_inst(args):
args = basic_setting(args)
args = basic_mean_teacher(args)
args = bs_128_setting(args)
args = mt_fst_ramp_down_setting(args)
args = mt_load_from_inst(args)
args.whichimagenet = 'part10'
args.expId = 'mt_part10_res18_inst'
return args
def imagenet_instance_setting_res101_3blk(args):
args = basic_setting(args)
args = basic_instance_setting(args)
args.valinum = 250
args.valbatchsize = 200
args.whichimagenet = 'full_widx'
args.network_func = 'get_resnet_101_3blk'
args.network_func_kwargs = '{"num_cat": 128}'
return args
def res18_ir_depth(args):
args = shared_sts.basic_setting(args)
args = basic_inst_setting(args)
args = save_setting(args)
args = shared_sts.bs_64_less_save_setting(args)
args.network_func = 'combine_irla_others.get_resnet18_inst_and_depth'
args.dataconfig = 'dataset_config_pbrimage.cfg'
args.expId = 'ir_depth'
args.lr_boundaries = '1300011'
return args
def infant_instance_setting(args):
args = basic_setting(args)
args = basic_instance_setting(args)
args.valinum = 123
args.valbatchsize = 100
args.whichimagenet = 'full_new_widx'
args.instance_data_len = 305228
args.network_func = 'get_resnet_18'
args.network_func_kwargs = '{"num_cat": 128}'
return args
def mean_teacher_res18(args):
args = basic_setting(args)
args = basic_mean_teacher(args)
args = bs_128_setting(args)
args.weight_decay = 1e-4
args.init_lr = 0.03
args.network_func = 'get_mean_teacher_resnet_18'
args.dbname = 'combine_instance'
args.expId = 'res18_mt_0'
return args
def res18_ir_cate_separate_bn(args):
args = shared_sts.basic_setting(args)
args = basic_la_cate_setting(args)
args = save_setting(args)
args = shared_sts.bs_256_less_save_setting(args)
args.network_func = 'get_resnet_18_cate_inst_branch2'
args.dataconfig = 'dataset_task_config_image_cate_ir_branch2.json'
args.expId = 'ir_cate_bn'
args.lr_boundaries = "500000,800000,960000"
args.ignorebname_new = 0
return args
def res18_ir(args):
args = shared_sts.basic_setting(args)
args = basic_inst_setting(args)
args = save_setting(args)
args = shared_sts.bs_128_less_save_setting(args)
args.network_func = 'get_resnet_18'
args.network_func_kwargs = '{"num_cat": 128}'
args.dataconfig = 'dataset_config_image.cfg'
args.expId = 'ir'
args.lr_boundaries = '1780011,2400011'
return args
def inst_and_mt_rmp_dwn_res18(args):
args = basic_setting(args)
args = basic_mean_teacher(args)
args = bs_128_setting(args)
args = basic_inst_setting(args)
args.dbname = 'combine_instance'
args.expId = 'inst_and_mt_rmp_dwn_res18'
args.network_func = 'get_mean_teacher_and_inst_resnet_18'
args.mt_ramp_down = 1
args.mt_ramp_down_epoch = 110
args.target_lr = 0.0125
return args
def res18_encode1to3_frm_cate(args):
args = shared_sts.basic_setting(args)
args = basic_inst_setting(args)
args = save_setting(args)
args = shared_sts.bs_128_less_save_setting(args)
args.dataconfig = 'dataset_config_image.cfg'
args.expId = 'encode1to3_frm_cate_fx'
args = load_from_inst_prep_res18_cate(args)
args.network_func = 'combine_irla_others.get_resnet_18_fx_encdX'
# Uncomment for beginning
args.load_param_dict = get_fx_encode1toX_param_dict_from_cate_model()
args.lr_boundaries = '1780011,2400011'
return args
def res18_la_cate(args):
args = shared_sts.basic_setting(args)
args = basic_la_cate_setting(args)
args = save_setting(args)
args = shared_sts.bs_256_less_save_setting(args)
args.dataconfig = 'dataset_task_config_image_la_cate.json'
args = load_from_res18_ir_cate(args)
args.loadstep = 50050
args.expId = 'la_cate'
# Uncomment for beginning
#args.load_param_dict = get_load_param_dict_ir2la()
args.lr_boundaries = '1040011,1550011'
return args
def inst_and_mean_teacher_res18(args):
args = basic_setting(args)
args = basic_mean_teacher(args)
args = bs_128_setting(args)
args = basic_inst_setting(args)
args.dbname = 'combine_instance'
args.expId = 'res18_inst_mt_cont'
args.network_func = 'get_mean_teacher_and_inst_resnet_18'
args.lr_boundaries = "480000"
args.loadexpId = 'res18_inst_mt_1'
args.loadstep = 480480
return args
def res18_la(args):
args = shared_sts.basic_setting(args)
args = basic_la_setting(args)
args = save_setting(args)
args = shared_sts.bs_128_less_save_setting(args)
args.dataconfig = 'dataset_task_config_image_la.json'
args = load_from_inst_prep_res18_ir(args)
args.loadstep = 100000
args.expId = 'la'
# Uncomment for beginning
#args.load_param_dict = get_load_param_dict_ir2la()
args.lr_boundaries = '1740011,2400011'
return args
def mt_part10_res50(args):
args = basic_setting(args)
args = basic_mean_teacher(args)
args = bs_128_setting(args)
args = mt_fst_ramp_down_setting(args)
args.whichimagenet = 'part10'
args.expId = 'mt_part10_res50_fx'
args.nport = 27006
args.network_func = 'get_mean_teacher_resnet_50'
args.mt_ramp_down_epoch = 120
args.fre_filter = 100100
args.fre_cache_filter = 10010
return args
def res18_encode1_frm_cat(args):
args = shared_sts.basic_setting(args)
args = basic_inst_setting(args)
args = save_setting(args)
args = shared_sts.bs_128_less_save_setting(args)
args.dataconfig = 'dataset_config_image.cfg'
args.expId = 'encode1_frm_cat'
args = load_from_inst_prep_res18_cate(args)
args.network_func = 'combine_irla_others.get_resnet_18_fx_encd1'
# Uncomment for beginning
#args.load_param_dict = '{"encode1/weights": "inst_encode1/weights", "encode1/bias": "inst_encode1/bias"}'
args.lr_boundaries = '1780011,2400011'
return args
def inst_and_cate_res18(args):
args = basic_setting(args)
args.init_type = 'variance_scaling_initializer'
args.tpu_flag = 1
args.weight_decay = 1e-4
args.imgnt_w_idx = True
args.whichopt = 0
args.dataconfig = 'dataset_task_config_image_ir_cate.json'
args.init_lr = 0.03
args.lr_boundaries = "500000,800000,960000"
args.network_func = 'get_resnet_18_inst_and_cate'
args.valinum = 250
args.valbatchsize = 200
args.whichimagenet = 'full_widx'
args.batchsize = 256
args.dbname = 'combine_instance'
args.expId = 'res18_inst_cate_1'
args.fre_filter = 10010
args.fre_valid = 5005
return args
