Skip to content

Commit

Permalink
feat: add validate shuffle config and set default to False
Browse files Browse the repository at this point in the history
  • Loading branch information
ChongWei905 committed Jul 17, 2024
1 parent af7fd64 commit 1c283bc
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 0 deletions.
2 changes: 2 additions & 0 deletions config.py
Original file line number Diff line number Diff line change
Expand Up @@ -263,6 +263,8 @@ def create_parser():
help='Loss scale (default=1.0)')
group.add_argument('--drop_overflow_update', type=bool, default=False,
help='Whether to execute optimizer if there is an overflow (default=False)')
group.add_argument('--eval_shuffle', type=bool, default=False,
help='Whether to shuffle the evaluation data (default=False)')

return parser_config, parser
# fmt: on
Expand Down
1 change: 1 addition & 0 deletions validate.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ def validate(args):
split=args.val_split,
num_parallel_workers=args.num_parallel_workers,
download=args.dataset_download,
shuffle=args.eval_shuffle,
)

# create transform
Expand Down

0 comments on commit 1c283bc

Please sign in to comment.