(model, checkpoint_path)
| 52 | |
| 53 | |
| 54 | def load_wts(model, checkpoint_path): |
| 55 | ckpt = torch.load(checkpoint_path, map_location='cpu') |
| 56 | return load_state_dict(model, ckpt) |
| 57 | |
| 58 | |
| 59 | def load_state_dict_from_url(model, url, **kwargs): |
no test coverage detected