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

Update main.py (#2872)

parent 756fdd8e
......@@ -113,8 +113,8 @@ def main():
help='input batch size for training (default: 256)')
parser.add_argument('--epochs', type=int, default=100,
help='number of epochs to train (default: 100)')
parser.add_argument('--num_workers', type=int, default=4,
help='number of workers (default: 4)')
parser.add_argument('--num_workers', type=int, default=0,
help='number of workers (default: 0)')
parser.add_argument('--log_dir', type=str, default="",
help='tensorboard log directory. If not specified, '
'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