From 96b8c28e5c4cc400a27b1ea69ddaac238eed5335 Mon Sep 17 00:00:00 2001 From: Andre Anjos <andre.dos.anjos@gmail.com> Date: Tue, 28 May 2024 16:16:52 +0200 Subject: [PATCH] [scripts.train] Fix description --- src/mednet/scripts/train.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/mednet/scripts/train.py b/src/mednet/scripts/train.py index 8ae78f79..9c25de7a 100644 --- a/src/mednet/scripts/train.py +++ b/src/mednet/scripts/train.py @@ -86,12 +86,11 @@ def reusable_options(f): "parameter, used in conjunction with the batch-size, may be used to " "reduce the number of samples loaded in each iteration, to affect memory " "usage in exchange for processing time (more iterations). This is " - "especially interesting when one is training on GPUs with a limited amount " - "of onboard RAM. processing time (more iterations). This is especially interesting " - "when one is training on GPUs with limited RAM. The default of 1 forces " - "the whole batch to be processed at once. Otherwise the batch is " - "multiplied by accumulate-grad-batches pieces, and gradients are accumulated " - "to complete each step.", + "useful interesting when one is training on GPUs with a limited amount " + "of onboard RAM. The default of 1 forces the whole batch to be " + "processed at once. Otherwise the batch is multiplied by " + "accumulate-grad-batches pieces, and gradients are accumulated " + "to complete each training step.", required=True, show_default=True, default=1, -- GitLab