yuxindu commited on
Commit
5a1171a
·
1 Parent(s): ef9b4bd

weight file path

Browse files
Files changed (1) hide show
  1. 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 = 'model/asset/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)
 
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)