Commit 0528c591 authored by lishen's avatar lishen

[fix]

parent 6496262f
......@@ -24,7 +24,7 @@ def parse_option():
parser.add_argument("--test_path", type=str, default="E:/datasets/food101/meta_data/test_full.txt",
help='path to testing list')
parser.add_argument('--weight_path', default="E:/Pretrained_model/food2k_resnet50_0.0001.pth", help='path to the pretrained model')
parser.add_argument('--use_checkpoint', action='store_true', default=True,
parser.add_argument('--use_checkpoint', action='store_true', default=False,
help="whether to use gradient checkpointing to save memory")
parser.add_argument('--checkpoint', type=str, default="E:/Pretrained_model/model.pth",
help="the path to checkpoint")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment