Predictor¶

The main module for run inference

class piepline.predict.Predictor(model: torch.nn.modules.module.Module, checkpoints_manager: piepline.utils.checkpoints_manager.CheckpointsManager, device: torch.device = None)[source]¶

Predictor run inference by training parameters

Args:: model (Module): model object, used for predict checkpoints_manager (CheckpointsManager): checkpoints manager device (torch.device or str): target device

predict(data: torch.Tensor)[source]¶

Predict one data batch

Parameters:	data – data as `torch.Tensor` or dict with key `data`
Returns:	processed output
Return type:	model output type