diff options
author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-08 11:45:37 +0800 |
---|---|---|
committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-08 11:45:37 +0800 |
commit | 57857a49c7fd95a9eaea208bb565df0e8b793149 (patch) | |
tree | ae51546967cf00b30623fa925b3a9a41b7dfd974 /simclr/main.py | |
parent | 9245e96df5e4eab16b001817d0e3973b356e113f (diff) |
Clarify how the iters calculated
Diffstat (limited to 'simclr/main.py')
-rw-r--r-- | simclr/main.py | 3 |
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)') |