(args, cfg)
| 77 | |
| 78 | |
| 79 | def build_model_main(args, cfg): |
| 80 | print(args.modelname) |
| 81 | from models.registry import MODULE_BUILD_FUNCS |
| 82 | assert args.modelname in MODULE_BUILD_FUNCS._module_dict |
| 83 | build_func = MODULE_BUILD_FUNCS.get(args.modelname) |
| 84 | model, criterion, postprocessors, _ = build_func( |
| 85 | args, cfg) |
| 86 | return model, criterion, postprocessors, _ |
| 87 | |
| 88 | |
| 89 | def main(args): |