Skip to content

Commit 659a908

Browse files
change gitignore, speed up inference for baseline
1 parent acf90bb commit 659a908

2 files changed

Lines changed: 5 additions & 2 deletions

File tree

.gitignore

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,9 @@
11
.old/
22
conversion/
33
CLAUDE.md
4-
4+
outputs/
5+
*.nii.gz
6+
v3/
57
# Created by https://www.toptal.com/developers/gitignore/api/visualstudiocode,python
68
# Edit at https://www.toptal.com/developers/gitignore?templates=visualstudiocode,python
79

src/baseline-inference/inference.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@
4545
MODEL_PATH = "weights/best_model.pth"
4646
PATCH_SIZE = (192,192,192)
4747
SW_BATCH = 2
48-
OVERLAP = 0.75
48+
OVERLAP = 0.5
4949

5050

5151
# -----------------------------
@@ -92,6 +92,7 @@
9292
model,
9393
overlap=OVERLAP,
9494
mode="gaussian",
95+
progress=True
9596
)
9697

9798
pred = pred.cpu().numpy()[0,0]

0 commit comments

Comments
 (0)