(args, cfg)
| 33 | |
| 34 | |
| 35 | def get_model(args, cfg): |
| 36 | model = build_model(cfg) |
| 37 | ckpt = torch.load(args.ckpt, map_location='cpu')['model'] |
| 38 | |
| 39 | model.load_state_dict(ckpt) |
| 40 | return model |
| 41 | |
| 42 | |
| 43 | def speed_test(model, input): |
no test coverage detected