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 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_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 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