Spaces:
Runtime error
Runtime error
Update model/run_inference.py
Browse files- 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}"
|