Skip to content

Commit d9626e4

Browse files
committed
Allow scaling through bash command
1 parent 3ec31ad commit d9626e4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

flamingo_tools/segmentation/unet_prediction.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -571,6 +571,8 @@ def run_unet_prediction_slurm(
571571
"""
572572
os.makedirs(output_folder, exist_ok=True)
573573
prediction_instances = int(prediction_instances)
574+
if isinstance(scale, str):
575+
scale = float(scale)
574576
slurm_task_id = os.environ.get("SLURM_ARRAY_TASK_ID")
575577

576578
if s3 is not None:

0 commit comments

Comments
 (0)