From 57857a49c7fd95a9eaea208bb565df0e8b793149 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
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(-)

(limited to 'simclr')

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