From 47fac9ff73aceedd267db1e734a98de122fc9430 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 16 Nov 2021 13:58:15 +0100 Subject: [PATCH] Update train.py comment to 'Model attributes' (#5670) --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 91bcd1e1e2e8..2838936d2d78 100644 --- a/train.py +++ b/train.py @@ -243,7 +243,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary if cuda and RANK != -1: model = DDP(model, device_ids=[LOCAL_RANK], output_device=LOCAL_RANK) - # Model parameters + # Model attributes nl = de_parallel(model).model[-1].nl # number of detection layers (to scale hyps) hyp['box'] *= 3 / nl # scale to layers hyp['cls'] *= nc / 80 * 3 / nl # scale to classes and layers