Predictor¶

The main module for run inference

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

Predictor run inference by training parameters

Parameters:	model – model object, used for predict fsm – `FileStructManager` object

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

Predict ine data

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