diff --git a/tools/train+sam.py b/tools/train+sam.py index 592c230..d0bfea8 100644 --- a/tools/train+sam.py +++ b/tools/train+sam.py @@ -22,7 +22,7 @@ import mmcv_custom def parse_args(): parser = argparse.ArgumentParser(description='Train a pose model') 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( '--resume-from', help='the checkpoint file to resume from')