File size: 1,218 Bytes
5de1792 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
def get_pretraining_set(opts):
if 'ntu' in opts.name:
from feeder.feeder_pretraining import Feeder
training_data = Feeder(**opts.train_feeder_args)
elif 'pkuv2' in opts.name:
from feeder.feeder_v2_pretrain import Feeder
training_data = Feeder(**opts.train_feeder_args)
return training_data
def get_finetune_training_set(opts):
if 'ntu' in opts.name:
from feeder.feeder_downstream import Feeder
data = Feeder(**opts.train_feeder_args)
elif 'pkuv2' in opts.name:
from feeder.feeder_v2_down import Feeder
data = Feeder(**opts.train_feeder_args)
elif 'pkuv1' in opts.name:
from feeder.feeder_v1_train import Feeder
data = Feeder(**opts.train_feeder_args)
return data
def get_finetune_validation_set(opts):
if 'ntu' in opts.name:
from feeder.feeder_downstream import Feeder
data = Feeder(**opts.test_feeder_args)
elif 'pkuv2' in opts.name:
from feeder.feeder_v2_down import Feeder
data = Feeder(**opts.test_feeder_args)
elif 'pkuv1' in opts.name:
from feeder.feeder_v1_val import Feeder
data = Feeder(**opts.test_feeder_args)
return data
|