Skip to content

Commit 69ae862

Browse files
ChongWei905ChongWei905
andauthored
feat: add validate shuffle config and set default to False (#793)
(cherry picked from commit 1c283bc) Co-authored-by: ChongWei905 <weichong4@huawei.com>
1 parent c744121 commit 69ae862

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

config.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -263,6 +263,8 @@ def create_parser():
263263
help='Loss scale (default=1.0)')
264264
group.add_argument('--drop_overflow_update', type=bool, default=False,
265265
help='Whether to execute optimizer if there is an overflow (default=False)')
266+
group.add_argument('--eval_shuffle', type=bool, default=False,
267+
help='Whether to shuffle the evaluation data (default=False)')
266268

267269
return parser_config, parser
268270
# fmt: on

validate.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ def validate(args):
3535
split=args.val_split,
3636
num_parallel_workers=args.num_parallel_workers,
3737
download=args.dataset_download,
38+
shuffle=args.eval_shuffle,
3839
)
3940

4041
# create transform

0 commit comments

Comments
 (0)