mirrors/deit
1
0
Fork 0
mirror of https://github.com/facebookresearch/deit.git synced 2025-06-03 14:52:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix argument bug

Browse Source
This commit is contained in:
fabfish 2023-05-28 16:52:55 +08:00
parent f17301d600
commit bd8fdf25fd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@ -181,7 +181,7 @@ def get_args_parser():
parser.set_defaults(pin_mem=True)
# distributed training parameters
parser.add_argument('--distributed', action='store_true', defaul=False, help='Enabling distributed training')
parser.add_argument('--distributed', action='store_true', default=False, help='Enabling distributed training')
parser.add_argument('--world_size', default=1, type=int,
help='number of distributed processes')
parser.add_argument('--dist_url', default='env://', help='url used to set up distributed training')