(self, filename)
| 27 | utils.torch_save(self, filename) |
| 28 | |
| 29 | def load(self, filename): |
| 30 | print(f"Loading image classifier from {filename}") |
| 31 | return utils.torch_load(self, filename) |
| 32 | |
| 33 | |
| 34 | class ClassificationHead(torch.nn.Linear): |
no outgoing calls
no test coverage detected