(flags=None, mode=None)
| 59 | |
| 60 | |
| 61 | def build_data_loader(flags=None, mode=None): |
| 62 | assert mode in ["train", "validation", "test"], "Nonsupport mode:{}".format(mode) |
| 63 | if mode == "train": |
| 64 | dataloader_infor = deepcopy(flags.TrainReader.dataloader) |
| 65 | elif mode == "validation": |
| 66 | dataloader_infor = deepcopy(flags.EvalReader.dataloader) |
| 67 | dataloader = create_module(dataloader_infor)(flags) |
| 68 | return dataloader |
| 69 | |
| 70 | |
| 71 | def build_optimizer(flags, model): |