weight file path
Browse files- model/inference_cpu.py +1 -1
model/inference_cpu.py
CHANGED
@@ -16,7 +16,7 @@ def set_parse():
|
|
16 |
# %% set up parser
|
17 |
parser = argparse.ArgumentParser()
|
18 |
parser.add_argument("--test_mode", default=True, type=bool)
|
19 |
-
parser.add_argument("--resume", type = str, default = '
|
20 |
parser.add_argument("-infer_overlap", default=0.0, type=float, help="sliding window inference overlap")
|
21 |
parser.add_argument("-spatial_size", default=(32, 256, 256), type=tuple)
|
22 |
parser.add_argument("-patch_size", default=(4, 16, 16), type=tuple)
|
|
|
16 |
# %% set up parser
|
17 |
parser = argparse.ArgumentParser()
|
18 |
parser.add_argument("--test_mode", default=True, type=bool)
|
19 |
+
parser.add_argument("--resume", type = str, default = 'SegVol_v1.pth')
|
20 |
parser.add_argument("-infer_overlap", default=0.0, type=float, help="sliding window inference overlap")
|
21 |
parser.add_argument("-spatial_size", default=(32, 256, 256), type=tuple)
|
22 |
parser.add_argument("-patch_size", default=(4, 16, 16), type=tuple)
|