Skip to content

Commit a9af0e2

Browse files
Remove Hindi links
1 parent c07b9a8 commit a9af0e2

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

classify/train.py

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@
7676

7777

7878
def train(opt, device):
79-
"""Trains a YOLOv5 model, managing datasets, model optimization, logging, and saving checkpoints."""
79+
"""Trains a YOLOv5 model on a dataset, managing data loading, optimization, logging, and checkpoint saving."""
8080
init_seeds(opt.seed + 1 + RANK, deterministic=True)
8181
save_dir, data, bs, epochs, nw, imgsz, pretrained = (
8282
opt.save_dir,
@@ -311,9 +311,7 @@ def lf(x):
311311

312312

313313
def parse_opt(known=False):
314-
"""Parses command line arguments for YOLOv5 training including model path, dataset, epochs, and more, returning
315-
parsed arguments.
316-
"""
314+
"""Parses command line arguments for YOLOv5 training, returning parsed arguments."""
317315
parser = argparse.ArgumentParser()
318316
parser.add_argument("--model", type=str, default="yolov5s-cls.pt", help="initial weights path")
319317
parser.add_argument("--data", type=str, default="imagenette160", help="cifar10, cifar100, mnist, imagenet, ...")
@@ -367,8 +365,17 @@ def main(opt):
367365
def run(**kwargs):
368366
"""
369367
Executes YOLOv5 model training or inference with specified parameters, returning updated options.
370-
371-
Example: from yolov5 import classify; classify.train.run(data=mnist, imgsz=320, model='yolov5m')
368+
369+
Args:
370+
**kwargs (Any): Arbitrary keyword arguments to override default training or inference options.
371+
372+
Returns:
373+
(argparse.Namespace): Updated options after parsing and applying overrides.
374+
375+
Examples:
376+
Run YOLOv5 training with custom parameters
377+
>>> from yolov5 import classify
378+
>>> classify.train.run(data='mnist', imgsz=320, model='yolov5m')
372379
"""
373380
opt = parse_opt(True)
374381
for k, v in kwargs.items():

0 commit comments

Comments
 (0)