From cb4cea561a3f39bcd6a3105c72d7e0b2b928bf44 Mon Sep 17 00:00:00 2001
From: Josua Rieder <josua.rieder@protonmail.com>
Date: Mon, 20 Jan 2025 00:56:50 +0100
Subject: [PATCH] add arguments to the respective argument groups

---
 train.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/train.py b/train.py
index 6e7057bc..fda3c184 100755
--- a/train.py
+++ b/train.py
@@ -81,17 +81,17 @@ group = parser.add_argument_group('Dataset parameters')
 # Keep this argument outside the dataset group because it is positional.
 parser.add_argument('data', nargs='?', metavar='DIR', const=None,
                     help='path to dataset (positional is *deprecated*, use --data-dir)')
-parser.add_argument('--data-dir', metavar='DIR',
+group.add_argument('--data-dir', metavar='DIR',
                     help='path to dataset (root dir)')
-parser.add_argument('--dataset', metavar='NAME', default='',
+group.add_argument('--dataset', metavar='NAME', default='',
                     help='dataset type + name ("<type>/<name>") (default: ImageFolder or ImageTar if empty)')
 group.add_argument('--train-split', metavar='NAME', default='train',
                    help='dataset train split (default: train)')
 group.add_argument('--val-split', metavar='NAME', default='validation',
                    help='dataset validation split (default: validation)')
-parser.add_argument('--train-num-samples', default=None, type=int,
+group.add_argument('--train-num-samples', default=None, type=int,
                     metavar='N', help='Manually specify num samples in train split, for IterableDatasets.')
-parser.add_argument('--val-num-samples', default=None, type=int,
+group.add_argument('--val-num-samples', default=None, type=int,
                     metavar='N', help='Manually specify num samples in validation split, for IterableDatasets.')
 group.add_argument('--dataset-download', action='store_true', default=False,
                    help='Allow download of dataset for torch/ and tfds/ datasets that support it.')
@@ -184,7 +184,7 @@ group.add_argument('--no-ddp-bb', action='store_true', default=False,
 group.add_argument('--synchronize-step', action='store_true', default=False,
                    help='torch.cuda.synchronize() end of each step')
 group.add_argument("--local_rank", default=0, type=int)
-parser.add_argument('--device-modules', default=None, type=str, nargs='+',
+group.add_argument('--device-modules', default=None, type=str, nargs='+',
                     help="Python imports for device backend modules.")
 
 # Optimizer parameters