Example #1
0
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
Example #2
0
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
Example #3
0
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
Example #4
0
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