Mercurial > repos > goeckslab > image_learner

diff image_learner.xml @ 21:d5c582cf74bc draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/goeckslab/gleam.git commit eed8c1e1d99a8a0c8f3a6bfdd8af48a5bfa19444
author goeckslab
date Tue, 20 Jan 2026 01:25:35 +0000
parents 64872c48a21f
children
line wrap: on
line diff
--- a/image_learner.xml	Tue Jan 06 15:35:11 2026 +0000
+++ b/image_learner.xml	Tue Jan 20 01:25:35 2026 +0000
@@ -28,6 +28,8 @@
             #set $sanitized_input_csv = re.sub('[^\w\-_\.]', '_', $input_csv.element_identifier.strip())
             ln -sf '$input_csv' "./${sanitized_input_csv}";
             #end if
+            #set $TORCH_HOME = "./torch_cache";
+            mkdir -p "$TORCH_HOME";
 
             #set $selected_validation_metric = ""
             #if $task_selection.task == "binary"
@@ -89,6 +91,7 @@
                 #end if
                 --image-resize "$image_resize"
                 --random-seed "$random_seed"
+                --torch-home "$TORCH_HOME"
                 --output-dir "." &&
 
             mkdir -p '$output_model.extra_files_path' &&