Browse Source

Merge branch 'main' of http://111.229.203.214:3000/fhw/ViTPose into main

main
fhw 1 year ago
parent
commit
05b2b71868
  1. 2
      tools/train+sam.py

2
tools/train+sam.py

@ -22,7 +22,7 @@ import mmcv_custom
def parse_args(): def parse_args():
parser = argparse.ArgumentParser(description='Train a pose model') parser = argparse.ArgumentParser(description='Train a pose model')
parser.add_argument('config', help='train config file path') parser.add_argument('config', help='train config file path')
parser.add_argument('-c', '--checkpoint', help='checkpoint file', default='/root/autodl-tmp/code/ViTPose/checkpoints/vitpose/vitpose-b.pth') parser.add_argument('-c', '--checkpoint', help='checkpoint file', default='checkpoints/vitpose/vitpose-b.pth')
parser.add_argument('--work-dir', help='the dir to save logs and models') parser.add_argument('--work-dir', help='the dir to save logs and models')
parser.add_argument( parser.add_argument(
'--resume-from', help='the checkpoint file to resume from') '--resume-from', help='the checkpoint file to resume from')

Loading…
Cancel
Save