From d049e22769bc3ba0ece62aa8b42ed96bd089411d Mon Sep 17 00:00:00 2001
From: Mahdi Amrollahi <44016758+M-Amrollahi@users.noreply.github.com>
Date: Thu, 7 Nov 2024 02:33:15 +0330
Subject: [PATCH] Update kfold-cross-validation.md (#17332)

Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
---
 docs/en/guides/kfold-cross-validation.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docs/en/guides/kfold-cross-validation.md b/docs/en/guides/kfold-cross-validation.md
index 381aef0e..44ba8d82 100644
--- a/docs/en/guides/kfold-cross-validation.md
+++ b/docs/en/guides/kfold-cross-validation.md
@@ -263,6 +263,7 @@ fold_lbl_distrb.to_csv(save_path / "kfold_label_distribution.csv")
 
     for k in range(ksplit):
         dataset_yaml = ds_yamls[k]
+        model = YOLO(weights_path, task="detect")
         model.train(data=dataset_yaml, epochs=epochs, batch=batch, project=project)  # include any train arguments
         results[k] = model.metrics  # save output metrics for further analysis
     ```