(flags)
| 51 | |
| 52 | |
| 53 | def build_model(flags): |
| 54 | # build network |
| 55 | model_infor = flags.Architecture.function |
| 56 | print('model_infor', model_infor) |
| 57 | model = create_module(model_infor)(flags) |
| 58 | return model |
| 59 | |
| 60 | |
| 61 | def build_data_loader(flags=None, mode=None): |
no test coverage detected