def res18_cpc_imagenet(args):
args = basic_setting(args)
args = save_setting(args)
args = bs_128_less_save_setting(args)
args = cpc_setting(args)
args.expId = 'cpc'
return args
def res18_dp_pbr(args):
args = basic_setting(args)
args = save_setting(args)
args = bs_64_setting(args)
args = dp_pbr_setting(args)
args.expId = 'dp_pbr'
args.lr_boundaries = '639998,1000011,1539988'
return args
def res18_cpc_s0(args):
args = basic_setting(args)
args = save_setting(args)
args = bs_128_less_save_setting(args)
args = cpc_setting(args)
args.expId = 'cpc_s0'
args.train_num_steps = 1310000
return args
def res18_AE_imagenet(args):
args = basic_setting(args)
args = save_setting(args)
args = bs_128_less_save_setting(args)
args = ae_setting(args)
args.expId = 'ae'
args.lr_boundaries = '660011,1100011'
return args
def res18_rp_imagenet(args):
args = basic_setting(args)
args = save_setting(args)
args = bs_64_setting(args)
args = rp_setting(args)
args.expId = 'rp_imgnt_fx'
args.lr_boundaries = '779998,1259988'
return args
def res18_col_imagenet(args):
args = basic_setting(args)
args = save_setting(args)
args = bs_64_setting(args)
args = col_setting(args)
args.expId = 'col_imgnt'
args.lr_boundaries = None
return args
def res18_ae_s0(args):
args = basic_setting(args)
args = save_setting(args)
args = bs_128_less_save_setting(args)
args = ae_setting(args)
args.expId = 'ae_s0'
args.lr_boundaries = '660011,1100011'
args.train_num_steps = 1210000
return args
def res18_rp_s0(args):
args = basic_setting(args)
args = save_setting(args)
args = bs_64_setting(args)
args = rp_setting(args)
args.lr_boundaries = '779998,1259988'
args.train_num_steps = 1359988
args.expId = 'rp_s0'
return args
def mt_part10_res50(args):
args = basic_setting(args)
args = basic_mean_teacher(args)
args = bs_128_setting(args)
args = mt_semi_basic(args)
args.whichimagenet = 'part10'
args.expId = 'mt_part10_res50_fx'
args.network_func = 'get_mean_teacher_resnet_50'
return args
def mt_part3_res18_rep1(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_rep1'
args.nport = 27007
return args
def set_tpu_test_basic(args):
args = shared_sts.basic_setting(args)
args = shared_sts.bs_256_less_save_setting(args)
args.valinum = 500
args.valbatchsize = 100
args.dataconfig = 'dataset_config_image.cfg'
args.nport = 26001
args.dbname = 'cb_image_trans'
args.namefunc = 'l3_prednet'
return args
def res18_col_s0(args):
args = basic_setting(args)
args = save_setting(args)
args = bs_64_setting(args)
args = col_setting(args)
args.expId = 'col_s0'
args.lr_boundaries = None
args.train_num_steps = 301 * 10010
return args
def mt_part01_res18(args):
args = basic_setting(args)
args = basic_mean_teacher(args)
args = bs_128_setting(args)
args = mt_semi_basic(args)
args.whichimagenet = 'part1_balanced'
args.expId = 'mt_part01_res18'
args.network_func = 'get_mean_teacher_resnet_18'
args.dataconfig = 'dataset_config_image_wun_widx.cfg'
return args
def res18_AE_imagenet_visualize(args):
args = basic_setting(args)
args = save_setting(args)
args = load_from_save_setting(args)
args = bs_128_less_save_setting(args)
args = ae_setting(args)
args.loadexpId = 'ae'
args.expId = 'ae_vis'
args.with_feat = 1
args.valinum = 10
return args
def res18_cpc_imagenet_tpu(args):
args = basic_setting(args)
args = bs_128_setting(args)
args = cpc_setting(args)
args.nport = 27001
args.expId = 'cpc'
args.tpu_task = 'cpc'
args.cacheDirPrefix = 'gs://cx_visualmaster/'
args.valinum = 390
args.valbatchsize = 128
return args
def res18_ir_s0(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_task_config_image_ir.json'
args.expId = 'ir_s0'
args.lr_boundaries = '1780011,2400011'
args.train_num_steps = 2610011
return args
def transfer_basic(args):
args = shared_sts.basic_setting(args)
args = shared_sts.bs_256_less_save_setting(args)
args.dataconfig = 'dataset_config_image.cfg'
args.nport = 26001
args.dbname = 'cb_image_trans'
args.collname = 'res18'
args.trainable_scope = 'category_trans'
args.drop_global_step = True
args.lr_boundaries = '200000,500000,700000'
args.weight_decay = 1e-4
args.train_num_steps = 801000
return args
def res18_la_s0(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_res18_ir_s0(args)
args.expId = 'la_s0'
# Uncomment for beginning
args.load_param_dict = get_load_param_dict_ir2la()
args.lr_boundaries = '1740011,2400011'
args.train_num_steps = 2610011
return args
def mt_part10_res18_relearn(args):
args = basic_setting(args)
args = basic_mean_teacher(args)
args = bs_128_setting(args)
args = mt_semi_basic(args)
args.whichimagenet = 'part10'
args.expId = 'mt_part10_res18_relearn'
args.network_func = 'get_mean_teacher_resnet_18'
args.dataconfig = 'dataset_config_image_wun_widx.cfg'
args.loadexpId = 'mt_part10_res18'
args.drop_global_step = True
args.loadstep = 1001000
return args
def prednet_kinetics_l3_test(args):
args = shared_sts.basic_setting(args)
args = shared_sts.bs_256_less_save_setting(args)
args.valinum = 500
args.valbatchsize = 100
args.dataconfig = 'dataset_config_image.cfg'
args.nport = 26001
args.dbname = 'cb_image_trans'
args.expId = 'prednet_kinetics_l3_test_ep3'
args.namefunc = 'l3_prednet'
args.loadport = 26001
args.load_dbname = 'cb_image_trans'
args.load_collname = 'res18'
args.loadexpId = 'prednet_kinetics_l3'
args.loadstep = 5005 * 3
ckpt_file = '/mnt/fs4/chengxuz/.tfutils/localhost:26001/cb_image_trans/combinet/prednet_kinetics_l3_test_ep3/checkpoint-15015'
args = set_load_param_dict(args, ckpt_file)
return args
