bitorch_engine.utils.convert.quantize_linear_with_binary_linear_cuda

bitorch_engine.utils.convert.quantize_linear_with_binary_linear_cuda(module: Module, names_to_replace: Iterable[str], parent_name: str = '')

Replace all layers contained in names_to_replace within the given module with BinaryLinearCuda layers. :param module: the module which contains Linear layers :param names_to_replace: the list of layer names to be replaced :param parent_name: the name of the parent (usually empty when called directly) :return: the list of names of layers which were replaced