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.
+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.
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.
@@ -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)