def vd_slowfast_a4_IR():
args = {}
args = basic_fix(args)
args = slow_bs(args)
args = vd_basic(args)
args = slowfast_a4_basic(args)
args['exp_id'] = 'vd_slowfast_a4_IR'
args['task'] = 'IR'
return args
def vd_tsrn_IR():
args = {}
args = basic_fix(args)
args = slow_bs(args)
args = vd_basic(args)
args['exp_id'] = 'vd_tsrn_IR'
args['model_type'] = 'tsrn'
args['task'] = 'IR'
return args
def vd_trn_f4_IR():
args = {}
args = basic_fix(args)
args = slow_bs(args)
args = vd_basic(args)
args['exp_id'] = 'vd_trn_f4_IR'
args['model_type'] = 'trn'
args['task'] = 'IR'
args['trn_num_frames'] = 4
return args
def vd_tsrn_slow():
args = {}
args = basic_fix(args)
args = slow_bs(args)
args = vd_basic(args)
args = load_from_tsrn_slow_IR(args)
args = res18_la(args)
args['exp_id'] = 'vd_tsrn_slow'
args['model_type'] = 'tsrn_slow'
return args
def vd_trn_pret():
args = {}
args = basic_fix(args)
args = slow_bs(args)
args = vd_basic(args)
args = load_from_LA_final(args)
args = res18_la(args)
args['exp_id'] = 'vd_trn_pret'
args['model_type'] = 'trn'
args['lr_boundaries'] = '1460011,1570011'
return args
def vd_tsrn_f4():
args = {}
args = basic_fix(args)
args = slow_bs(args)
args = vd_basic(args)
args = load_from_tsrn_f4_IR(args)
args = res18_la(args)
args['exp_id'] = 'vd_tsrn_f4'
args['model_type'] = 'tsrn'
args['trn_num_frames'] = 4
return args
def vd_slowfast_a4():
args = {}
args = basic_fix(args)
args = slow_bs(args)
args = vd_basic(args)
args = load_from_slowfast_a4_IR(args)
args = res18_la(args)
args = slowfast_a4_basic(args)
args['exp_id'] = 'vd_slowfast_a4'
args['lr_boundaries'] = '1120011,1294998'
return args
def vd_slowfast_a4_test():
args = {}
args = basic_fix(args)
args = slow_bs(args)
args = vd_basic(args)
args = res18_la(args)
args = slowfast_a4_basic(args)
args['load_exp'] = 'vd_unsup_fx/dyn_clstr/vd_slowfast_a4'
args['exp_id'] = 'vd_slowfast_a4_test'
args['lr_boundaries'] = '1120011,1294998'
args['plot_val'] = True
args['pure_test'] = True
return args
def vd_slowfast_a4_test_noimg_mre():
args = {}
args = basic_fix(args)
args = slow_bs(args)
args = vd_basic(args)
args = res18_la(args)
args = slowfast_a4_basic(args)
args['load_exp'] = 'vd_unsup_fx/dyn_clstr/vd_slowfast_a4'
args['exp_id'] = 'vd_slowfast_a4_test_noimg_mre'
args['lr_boundaries'] = '1120011,1294998'
args['plot_val'] = True
args['pure_test'] = True
args['plot_val_no_image'] = True
args['test_no_frames'] = 10
args['val_num_workers'] = 40
return args
def vd_big_bin_basic(args):
args = vd_basic(args)
args['data_len'] = 473532
args['bin_interval'] = 130
return args
def vd_bin_basic(args):
args = vd_basic(args)
args['data_len'] = 1157650
args['bin_interval'] = 52
return args