def METHOD_NAME(
    cls, path: Path, device: Optional[Union[str, torch.device]] = None
) -> "PyTorchPredictor":
    predictor = super().METHOD_NAME(path)
    if device is not None:
        device = resolve_device(device)
        predictor.to(device)
    state_dict = torch.load(
        path / "prediction-net-state.pt",
        map_location=device,
    )
    predictor.prediction_net.load_state_dict(state_dict)
    return predictor