Method info

ultralytics/engine/model.py:406–427

Display model information. This method provides an overview or detailed information about the model, depending on the arguments passed. It can control the verbosity of the output. Args: detailed (bool): If True, shows detailed information about the model layers

(self, detailed: bool = False, verbose: bool = True, imgsz: int | list[int, int] = 640)

        torch.save({**self.ckpt, **updates}, filename)

    def info(self, detailed: bool = False, verbose: bool = True, imgsz: int | list[int, int] = 640):
        """Display model information.

        This method provides an overview or detailed information about the model, depending on the arguments
        passed. It can control the verbosity of the output.

        Args:
            detailed (bool): If True, shows detailed information about the model layers and parameters.
            verbose (bool): If True, prints the information and returns model summary. If False, returns None.
            imgsz (int | list[int, int]): Input image size used for FLOPs calculation.

        Returns:
            (tuple): A tuple containing the number of layers (int), number of parameters (int), number of gradients
                (int), and GFLOPs (float). Returns None if verbose is False.

        Examples:
            >>> model = Model("yolo26n.pt")
            >>> model.info()  # Prints model summary and returns tuple
            >>> model.info(detailed=True)  # Prints detailed info and returns tuple
        """
        self._check_is_pytorch_model()
        return self.model.info(detailed=detailed, verbose=verbose, imgsz=imgsz)

    def fuse(self) -> Model:
        """Fuse Conv2d and BatchNorm2d layers in the model for optimized inference.