unet.trainer.evaluator¶
Classes¶
Main class for evaluating the model. |
Module Contents¶
- class unet.trainer.evaluator.Evaluator(logger: logging.Logger, test_dataset: altametris.unet.dataset.dataset.SegrailsDataset, model: altametris.unet.model.unet.Unet, eval_config: dict, log_dir: str)¶
Main class for evaluating the model.
- logger¶
- test_dataset¶
- model¶
- eval_config¶
- log_dir¶
- device¶
- eval()¶
Evaluating pipeline.
- _validate(val_loader)¶
Validation pipeline.