From 57857a49c7fd95a9eaea208bb565df0e8b793149 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Mon, 8 Aug 2022 11:45:37 +0800 Subject: Clarify how the iters calculated --- simclr/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/simclr/main.py b/simclr/main.py index 4dfdcf9..93a2d83 100644 --- a/simclr/main.py +++ b/simclr/main.py @@ -37,7 +37,8 @@ def parse_args_and_config(): parser.add_argument('--seed', default=None, type=int, help='Random seed for reproducibility') parser.add_argument('--num-iters', default=23438, type=int, - help='Number of iters') + help='Number of iters (default is 50 epochs equiv., ' + 'around dataset_size * epochs / batch_size)') parser.add_argument('--config', type=argparse.FileType(mode='r'), help='Path to config file (optional)') -- cgit v1.2.3