make minimum nimg_per_epoch=2 in the gui

carsen-stringer · carsen-stringer · commit 5e37794b9436 · 2025-05-03T11:20:22.000-04:00
diff --git a/cellpose/gui/gui.py b/cellpose/gui/gui.py
@@ -1811,7 +1811,7 @@ def train_model(self, restore=None, normalize_params=None):
         self.new_model_path, train_losses = train.train_seg(
             self.model.net, train_data=self.train_data, train_labels=self.train_labels,
             normalize=normalize_params, min_train_masks=0,
-            save_path=save_path, nimg_per_epoch=max(8, len(self.train_data)),
+            save_path=save_path, nimg_per_epoch=max(2, len(self.train_data)),
             learning_rate=self.training_params["learning_rate"],
             weight_decay=self.training_params["weight_decay"],
             n_epochs=self.training_params["n_epochs"],
diff --git a/cellpose/io.py b/cellpose/io.py
@@ -63,6 +63,7 @@ def logger_setup(cp_path=".cellpose", logfile_name="run.log", stdout_file_replac
                     level=logging.INFO,
                     format="%(asctime)s [%(levelname)s] %(message)s",
                     handlers=handlers,
+                    force=True
     )
     logger = logging.getLogger(__name__)
     logger.info(f"WRITING LOG OUTPUT TO {log_file}")

Original file line number	Diff line number	Diff line change
`@@ -63,6 +63,7 @@ def logger_setup(cp_path=".cellpose", logfile_name="run.log", stdout_file_replac`
`63`	`63`	`level=logging.INFO,`
`64`	`64`	`format="%(asctime)s [%(levelname)s] %(message)s",`
`65`	`65`	`handlers=handlers,`
	`66`	`+ force=True`
`66`	`67`	`)`
`67`	`68`	`logger = logging.getLogger(__name__)`
`68`	`69`	`logger.info(f"WRITING LOG OUTPUT TO {log_file}")`