(args)
| 61 | |
| 62 | |
| 63 | def build_dataset_pretrain(args): |
| 64 | args.pretrain = False |
| 65 | dataset_names = args.dataset |
| 66 | if ' ' in args.dataset: |
| 67 | datasets_name = args.dataset.split(' ') |
| 68 | dataset = [] |
| 69 | for dataset_ in datasets_name: |
| 70 | args.dataset = dataset_ |
| 71 | dataset.append(build_dataset(args)) |
| 72 | else: |
| 73 | dataset = [build_dataset(args)] |
| 74 | args.pretrain = True |
| 75 | args.dataset = dataset_names |
| 76 | dataset_class = getattr(importlib.import_module("cogdl.datasets.gcc_data"), "PretrainDataset") |
| 77 | return dataset_class(args.dataset, [x.data for x in dataset]) |
| 78 | |
| 79 | |
| 80 | def build_dataset(args): |
no test coverage detected