From c6f573105d8701ae0a58f49fbdf4d75b5550aed6 Mon Sep 17 00:00:00 2001 From: fhw Date: Wed, 10 Jul 2024 14:53:01 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E6=9D=83=E9=87=8D=E8=B7=AF?= =?UTF-8?q?=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tools/train+sam.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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')