(self, filter: IFilter, results_dir: str = None)
| 16 | |
| 17 | class ModelImageVisualizer: |
| 18 | def __init__(self, filter: IFilter, results_dir: str = None): |
| 19 | self.filter = filter |
| 20 | self.results_dir = None if results_dir is None else Path(results_dir) |
| 21 | self.results_dir.mkdir(parents=True, exist_ok=True) |
| 22 | |
| 23 | def _clean_mem(self): |
| 24 | torch.cuda.empty_cache() |
nothing calls this directly
no outgoing calls
no test coverage detected