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