parser.add_argument('-s', '--stacks', default=8, type=int, metavar='N', help='Number of hourglasses to stack') parser.add_argument('--features', default=256, type=int, metavar='N', help='Number of features in the hourglass') parser.add_argument('-b', '--blocks', default=1, type=int, metavar='N', help='Number of residual modules at each location in the hourglass') parser.add_argument('--num-classes', default=16, type=int, metavar='N', help='Number of keypoints') ------ optional arguments: -h, --help show this help message and exit --arch ARCH, -a ARCH model architecture: hg | preresnet110 | preresnet1202 | preresnet20 | preresnet32 | preresnet44 | preresnet56 (default: resnet18) -s N, --stacks N Number of hourglasses to stack --features N Number of features in the hourglass -b N, --blocks N Number of residual modules at each location in the hourglass --num-classes N Number of keypoints -j N, --workers N number of data loading workers (default: 4) --epochs N number of total epochs to run --start-epoch N manual epoch number (useful on restarts) --train-batch N train batchsize --test-batch N test batchsize
