yaghi27 commited on
Commit
06206f7
·
1 Parent(s): 57195b4

Update model/run_inference.py

Browse files
Files changed (1) hide show
  1. model/run_inference.py +2 -2
model/run_inference.py CHANGED
@@ -38,8 +38,8 @@ from mmdet3d.apis import init_model, inference_mono_3d_detector
38
 
39
  # ─── 2) Paths & device ───────────────────────────────────────────────
40
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
41
- CONFIG = "detr3d_r101_gridmask.py"
42
- CHECKPOINT = "epoch_30.pth"
43
 
44
  assert os.path.isfile(CONFIG), f"Config not found: {CONFIG}"
45
  assert os.path.isfile(CHECKPOINT), f"Checkpoint not found: {CHECKPOINT}"
 
38
 
39
  # ─── 2) Paths & device ───────────────────────────────────────────────
40
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
41
+ CONFIG = "model/detr3d_r101_gridmask.py"
42
+ CHECKPOINT = "model/epoch_30.pth"
43
 
44
  assert os.path.isfile(CONFIG), f"Config not found: {CONFIG}"
45
  assert os.path.isfile(CHECKPOINT), f"Checkpoint not found: {CHECKPOINT}"