diff --git a/docs/en/integrations/mlflow.md b/docs/en/integrations/mlflow.md index 1e6a2e6bd5..9d204fc52e 100644 --- a/docs/en/integrations/mlflow.md +++ b/docs/en/integrations/mlflow.md @@ -86,7 +86,7 @@ Make sure that MLflow logging is enabled in Ultralytics settings. Usually, this mlflow server --backend-store-uri runs/mlflow ``` - This will start a local server at http://127.0.0.1:5000 by default and save all mlflow logs to the 'runs/mlflow' directory. To specify a different URI, set the `MLFLOW_TRACKING_URI` environment variable. + This will start a local server at `http://127.0.0.1:5000` by default and save all mlflow logs to the 'runs/mlflow' directory. To specify a different URI, set the `MLFLOW_TRACKING_URI` environment variable. 4. **Kill MLflow Server Instances**: To stop all running MLflow instances, run: @@ -102,7 +102,7 @@ The logging is taken care of by the `on_pretrain_routine_end`, `on_fit_epoch_end 1. **Logging Custom Metrics**: You can add custom metrics to be logged by modifying the `trainer.metrics` dictionary before `on_fit_epoch_end` is called. -2. **View Experiment**: To view your logs, navigate to your MLflow server (usually http://127.0.0.1:5000) and select your experiment and run. YOLO MLflow Experiment +2. **View Experiment**: To view your logs, navigate to your MLflow server (usually `http://127.0.0.1:5000`) and select your experiment and run. YOLO MLflow Experiment 3. **View Run**: Runs are individual models inside an experiment. Click on a Run and see the Run details, including uploaded artifacts and model weights. YOLO MLflow Run @@ -188,7 +188,7 @@ To start an MLflow server for tracking your experiments in Ultralytics YOLO, use mlflow server --backend-store-uri runs/mlflow ``` -This command starts a local server at http://127.0.0.1:5000 by default. If you need to stop running MLflow server instances, use the following bash command: +This command starts a local server at `http://127.0.0.1:5000` by default. If you need to stop running MLflow server instances, use the following bash command: ```bash ps aux | grep 'mlflow' | grep -v 'grep' | awk '{print $2}' | xargs kill -9 diff --git a/docs/en/modes/train.md b/docs/en/modes/train.md index 2f3845d62f..8a19f95256 100644 --- a/docs/en/modes/train.md +++ b/docs/en/modes/train.md @@ -260,7 +260,7 @@ To use TensorBoard in [Google Colab](https://colab.research.google.com/github/ul tensorboard --logdir ultralytics/runs # replace with 'runs' directory ``` -To use TensorBoard locally run the below command and view results at . +To use TensorBoard locally run the below command and view results at `http://localhost:6006/`. !!! example diff --git a/ultralytics/nn/tasks.py b/ultralytics/nn/tasks.py index 97d52212a5..b8b47b834f 100644 --- a/ultralytics/nn/tasks.py +++ b/ultralytics/nn/tasks.py @@ -912,6 +912,9 @@ class YOLOEModel(DetectionModel): names (List[str]): List of class names. embeddings (torch.Tensor): Embeddings tensor. """ + assert not hasattr(self.model[-1], "lrpc"), ( + "Prompt-free model does not support setting classes. Please try with Text/Visual prompt models." + ) assert embeddings.ndim == 3 self.pe = embeddings self.model[-1].nc = len(names)