def ir_s1(args):
args = transfer_basic(args)
load_func = 'combine_irla_others.res18_ir_s1'
args, train_args = load_prev_func(args, load_func)
args.network_func = lambda: trans_network_cfg(train_args)
args.expId = 'ir_s1'
args.init_lr = 0.01
return args
def ae_s2(args):
args = transfer_basic(args)
load_func = 'other_tasks.res18_AE_imagenet_seed2'
args, train_args = load_prev_func(args, load_func)
args.network_func = lambda: trans_network_cfg(train_args)
args.expId = 'ae_s2'
args.init_lr = 0.01
return args
def untrn_s2(args):
args = transfer_basic(args)
load_func = 'cate_res18_exp2'
args, train_args = load_prev_func(args, load_func)
args.network_func = lambda: trans_network_cfg(train_args)
args = clear_load_params(args)
args.expId = 'untrn_s2'
args.init_lr = 0.01
return args
def cpc_s2(args):
args = transfer_basic(args)
load_func = 'other_tasks.res18_cpc_imagenet_tpu'
args, train_args = load_prev_func(args, load_func)
args.network_func = lambda: trans_network_cfg(train_args)
args = clear_load_params(args)
step_num = 130 * 10010
#args.ckpt_file = '/mnt/fs4/chengxuz/tpu_ckpts/cpc_seed2/model.ckpt-%i' % step_num
args.expId = 'cpc_s2'
args.init_lr = 0.01
return args