mirror of https://github.com/facebookresearch/deit
parent
0282d2a175
commit
dcd888df20
2
main.py
2
main.py
|
@ -128,7 +128,7 @@ def get_args_parser():
|
|||
help='How to apply mixup/cutmix params. Per "batch", "pair", or "elem"')
|
||||
|
||||
# Dataset parameters
|
||||
parser.add_argument('--data-path', default='/datasets01_101/imagenet_full_size/061417/', type=str,
|
||||
parser.add_argument('--data-path', default='/datasets01/imagenet_full_size/061417/', type=str,
|
||||
help='dataset path')
|
||||
parser.add_argument('--data-set', default='IMNET', choices=['CIFAR', 'IMNET', 'INAT', 'INAT19'],
|
||||
type=str, help='Image Net dataset path')
|
||||
|
|
Loading…
Reference in New Issue