예제 #1
0
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
예제 #2
0
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
예제 #3
0
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
예제 #4
0
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
예제 #5
0
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
예제 #6
0
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