Skip to content

GitLab

Unverified Commit 2ae891a7 authored by Mufei Li's avatar Mufei Li Committed by GitHub
Browse files

Update main.py (#2872)

parent 756fdd8e
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/pytorch/ogb_lsc/PCQM4M/main.py
View file @ 2ae891a7
...@@ -113,8 +113,8 @@ def main(): ...@@ -113,8 +113,8 @@ def main():
help='input batch size for training (default: 256)') help='input batch size for training (default: 256)')
parser.add_argument('--epochs', type=int, default=100, parser.add_argument('--epochs', type=int, default=100,
help='number of epochs to train (default: 100)') help='number of epochs to train (default: 100)')
parser.add_argument('--num_workers', type=int, default=4, parser.add_argument('--num_workers', type=int, default=0,
help='number of workers (default: 4)') help='number of workers (default: 0)')
parser.add_argument('--log_dir', type=str, default="", parser.add_argument('--log_dir', type=str, default="",
help='tensorboard log directory. If not specified, ' help='tensorboard log directory. If not specified, '
'tensorboard will not be used.') 'tensorboard will not be used.')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment