Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

✏️ fix typo

Browse Source
pull/1525/head
NPU-Franklin 2 years ago
parent 4201c718e9
commit 9152b10478
No known key found for this signature in database
GPG Key ID: C4EB0C928370B609
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
train.py
Unescape View File

@ -97,7 +97,7 @@ group.add_argument('--class-map', default='', type=str, metavar='FILENAME',
# Model parameters # Model parameters
group = parser.add_argument_group('Model parameters') group = parser.add_argument_group('Model parameters')
group.add_argument('--model', default='resnet50', type=str, metavar='MODEL', group.add_argument('--model', default='resnet50', type=str, metavar='MODEL',
help='Name of model to train (default: "resnet50"') help='Name of model to train (default: "resnet50")')
group.add_argument('--pretrained', action='store_true', default=False, group.add_argument('--pretrained', action='store_true', default=False,
help='Start with pretrained version of specified network (if avail)') help='Start with pretrained version of specified network (if avail)')
group.add_argument('--initial-checkpoint', default='', type=str, metavar='PATH', group.add_argument('--initial-checkpoint', default='', type=str, metavar='PATH',

Write Preview
Loading…
Cancel
Save