fix: up the vram requirement for nn_laser label studio

UCSD-E4E · Nov 13, 2024 · 708a2ce · 708a2ce
1 parent 7d1a8b4
commit 708a2ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fishsense_lite/commands/label_studio.py b/fishsense_lite/commands/label_studio.py
@@ -29,7 +29,7 @@
 from fishsense_lite.utils import get_output_file, get_root, uint16_2_uint8
 
 
-@ray.remote(vram_mb=768)
+@ray.remote(vram_mb=900)
 def execute_nn_laser(
     input_file: Path,
     lens_calibration: LensCalibration,