[good] uploading training script that works?

uhlmanngroup · Sep 30, 2024 · 2d3d546 · 2d3d546
1 parent 2d37b11
commit 2d3d546
Show file tree

Hide file tree

Showing 10 changed files with 1 addition and 1 deletion.
diff --git a/.env b/.env
diff --git a/bioimage_embed/inference.py b/bioimage_embed/inference.py
diff --git a/bioimage_embed/lightning/tests/__init__.py b/bioimage_embed/lightning/tests/__init__.py
diff --git a/bioimage_embed/models/legacy/__init__.py b/bioimage_embed/models/legacy/__init__.py
diff --git a/bioimage_embed/models/tests/legacy.py b/bioimage_embed/models/tests/legacy.py
diff --git a/bioimage_embed/models/tests/mae.py b/bioimage_embed/models/tests/mae.py
diff --git a/bioimage_embed/models/vit/tests/mae.py b/bioimage_embed/models/vit/tests/mae.py
diff --git a/bioimage_embed/tests/__init__.py b/bioimage_embed/tests/__init__.py
diff --git a/bioimage_embed/tests/test_training.py b/bioimage_embed/tests/test_training.py
diff --git a/scripts/idr/study.submitit.py b/scripts/idr/study.submitit.py
@@ -87,7 +87,6 @@ def __getitem__(self, idx):
         if torch.is_tensor(idx):
             idx = idx.tolist()
         img_name = self.file_list[idx]
-        obj = fs.open(img_name,filecache={'cache_storage':'tmp/idr'})
         try:
             image = read_image(fs,img_name)
             if self.transform:
@@ -190,6 +189,7 @@ def train(name, num_gpus_per_node=1,num_nodes=1):
         callbacks=None,
         # default_root_dir=f"lightning_logs/{name}",
         # plugin=[],
+
         logger=[wandb],
         accumulate_grad_batches=16,
         max_epochs=params["max_epochs"],