aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2022-08-08 11:45:37 +0800
committerJordan Gong <jordan.gong@protonmail.com>2022-08-08 11:45:37 +0800
commit57857a49c7fd95a9eaea208bb565df0e8b793149 (patch)
treeae51546967cf00b30623fa925b3a9a41b7dfd974
parent9245e96df5e4eab16b001817d0e3973b356e113f (diff)
Clarify how the iters calculated
-rw-r--r--simclr/main.py3
1 files changed, 2 insertions, 1 deletions
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)')