Skip to content

Commit

Permalink
change mlperf to lower
Browse files Browse the repository at this point in the history
  • Loading branch information
Ubuntu committed Oct 18, 2024
1 parent 4f3f355 commit 8c21ef4
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions examples/graphbolt/pyg/hetero/node_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -345,10 +345,10 @@ def parse_args():
"igb-het-small",
"igb-het-medium",
"igb-het-large",
"igb-het-MLPerf",
"igb-het-mlperf",
"igb-het",
],
help="Dataset name. Possible values: ogb-lsc-mag240m, igb-het, and igb-het-[tiny|small|medium|large|MLPerf].",
help="Dataset name. Possible values: ogb-lsc-mag240m, igb-het, and igb-het-[tiny|small|medium|large|mlperf].",
)
parser.add_argument(
"--fanout",
Expand Down
4 changes: 2 additions & 2 deletions examples/graphbolt/rgcn/hetero_rgcn.py
Original file line number Diff line number Diff line change
Expand Up @@ -671,10 +671,10 @@ def main(args):
"igb-het-medium",
"igb-het-large",
"igb-het",
"igb-het-MLPerf",
"igb-het-mlperf",
],
help="Dataset name. Possible values: ogbn-mag, ogb-lsc-mag240m, "
"igb-het, and igb-het-[tiny|small|medium|large|MLPerf].",
"igb-het, and igb-het-[tiny|small|medium|large|mlperf].",
)
parser.add_argument("--num_epochs", type=int, default=3)
parser.add_argument("--num_workers", type=int, default=0)
Expand Down
4 changes: 2 additions & 2 deletions python/dgl/graphbolt/impl/ondisk_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -1052,8 +1052,8 @@ class BuiltinDataset(OnDiskDataset):
"igb-het-large-seeds",
"igb-het",
"igb-het-seeds",
"igb-het-MLPerf",
"igb-het-MLPerf-seeds",
"igb-het-mlperf",
"igb-het-mlperf-seeds",
]
_all_datasets = _datasets + _large_datasets

Expand Down

0 comments on commit 8c21ef4

Please sign in to comment.