Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion megatron/training.py
Original file line number Diff line number Diff line change
Expand Up @@ -1071,7 +1071,7 @@ def build_train_valid_test_data_iterators(
if test_ds is not None else []

# Flags to know if we need to do training/validation/testing.
do_train = (train_dataloader is not None and args.train_iters > 0) and not args.eval_only
do_train = train_dataloader is not None and args.train_iters > 0 and not args.eval_only

# Need to broadcast num_tokens and num_type_tokens.
flags = torch.cuda.LongTensor([
Expand Down