Skip to content

Commit 936476b

Browse files
authored
patch size change (#51)
1 parent 9e0404a commit 936476b

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

  • src/unicorn_baseline/vision/radiology

src/unicorn_baseline/vision/radiology/main.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -234,6 +234,7 @@ def run_radiology_vision_task(
234234
model_dir=model_dir,
235235
domain=domain,
236236
title=image_input["interface"]["slug"],
237+
patch_size = [16, 64, 64],
237238
)
238239
neural_representations.append(neural_representation)
239240

@@ -248,6 +249,7 @@ def run_radiology_vision_task(
248249
model_dir=model_dir,
249250
domain=domain,
250251
title=image_input["interface"]["slug"],
252+
patch_size = [16, 128, 128],
251253
)
252254
neural_representations.append(neural_representation)
253255

0 commit comments

Comments
 (0)