filename call in main.go

htem · Nov 27, 2023 · c631d1a · c631d1a
1 parent 501bad6
commit c631d1a
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/optoseg/src/main.go b/optoseg/src/main.go
@@ -7,11 +7,10 @@ func main() {
         modelType   string
         scriptPath  string
         args        []string
-    }{
-        {"MTLSD", "./scripts/mtlsd_train.sh", []string{"--iterations", "100000", "--raw_file", "path/to/zarr/or/n5", "--voxel_size", "33"}},
-        {"ACLSD", "./scripts/aclsd_train.sh", []string{"--iterations", "100000", "--raw_file", "path/to/zarr/or/n5", "--warmup", "100000"}},
-        {"STELARR", "./scripts/stelarr_train.sh", []string{"--iterations", "100000", "--raw_file", "path/to/zarr/or/n5", "--warmup", "100000"}},
-        // Add more parameter sets as needed
+    }{ // TODO: add model type arg
+        {"MTLSD", "../batch_run.py", []string{"--iterations", "100000", "--raw_file", "path/to/zarr/or/n5", "--voxel_size", "33"}},
+        {"ACLSD", "../batch_run.py", []string{"--iterations", "100000", "--raw_file", "path/to/zarr/or/n5", "--warmup", "100000"}},
+        {"STELARR", "../batch_run.py", []string{"--iterations", "100000", "--raw_file", "path/to/zarr/or/n5", "--warmup", "100000"}},
     }
 
     autoseg.RunTrainingInParallel(trainingParams)