From c631d1a9bd1d35c56b5bd7e8ed5e7642b9422ef8 Mon Sep 17 00:00:00 2001 From: brianreicher Date: Mon, 27 Nov 2023 16:05:57 -0500 Subject: [PATCH] filename call in main.go --- optoseg/src/main.go | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/optoseg/src/main.go b/optoseg/src/main.go index c45c131..9b63749 100644 --- 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)