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
def depth_s0(args):
args = transfer_basic(args)
args = depth_network_cfg_setting(args)
args = clear_load_params(args)
step_num = 2982682
#args.ckpt_file = '/mnt/fs4/chengxuz/tpu_ckpts/depth_res18/model.ckpt-%i' % step_num
args.ignorebname_new = 0
args.add_batchname = '_pbrnet'
args.no_prep = 1
args.expId = 'depth_s0'
args.init_lr = 0.01
return args
def color_s0(args):
args = transfer_basic(args)
args = depth_network_cfg_setting(args)
args = clear_load_params(args)
step_num = 5605040
#args.ckpt_file = '/mnt/fs4/chengxuz/tpu_ckpts/col_res18/model.ckpt-%i' % step_num
args.ignorebname_new = 0
args.add_batchname = '_colorization'
args.color_dp_tl = 1
args.no_prep = 1
args.expId = 'color_s0'
args.init_lr = 0.01
return args
def rp_s0(args):
args = transfer_basic(args)
train_args = cmd_parser.get_parser().parse_args([])
train_args.pathconfig = 'rp_resnet18.cfg'
args.network_func = lambda: trans_network_cfg(train_args)
args = clear_load_params(args)
step_num = 1181162
#args.ckpt_file = '/mnt/fs4/chengxuz/tpu_ckpts/rp_res18/model.ckpt-%i' % step_num
args.ignorebname_new = 0
args.add_batchname = '_rp'
args.rp_dp_tl = 1
args.no_prep = 1
args.expId = 'rp_s0'
args.init_lr = 0.01
return args