lczerolens.adapt.builder#

AutoBuilder#

Model Wrapper#

class lczerolens.ModelWrapper(model: Module)#

Class for wrapping the LCZero models.

property device#: Returns the device.

forward(x)#: Forward pass.

classmethod from_onnx_path(onnx_model_path: str, check: bool = True)#: Builds a model from a given path.

classmethod from_path(model_path: str)#: Creates a wrapper from a model path.

classmethod from_torch_path(torch_model_path: str)#: Builds a model from a given path.

predict(to_pred: Board | Iterable[Board], with_grad: bool = False, input_requires_grad: bool = False, return_input: bool = False)#: Predicts the move.