(config, model, checkpoint_path )
| 96 | |
| 97 | |
| 98 | def load_model(config, model, checkpoint_path ): |
| 99 | checkpoint = torch.load(checkpoint_path, map_location='cpu') |
| 100 | model.load_state_dict(checkpoint['model']) |
| 101 | print(f'Model loaded with {checkpoint_path}') |
| 102 | |
| 103 | def patchify(imgs, patch_size): |
| 104 | """ |
nothing calls this directly
no test coverage detected