Skip to content

Commit

Permalink
fixing min_train_masks to 1 for GUI (#627)
Browse files Browse the repository at this point in the history
  • Loading branch information
carsen-stringer committed Jan 10, 2023
1 parent 06f42ef commit 9aff434
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cellpose/gui/gui.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,6 @@
except:
MATPLOTLIB = False


try:
from google.cloud import storage
os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = os.path.join(os.path.dirname(os.path.realpath(__file__)),
Expand Down Expand Up @@ -1643,6 +1642,7 @@ def train_model(self):
print('GUI_INFO: name of new model: ' + self.training_params['model_name'])
self.new_model_path = self.model.train(self.train_data, self.train_labels,
channels=self.channels,
min_train_masks=1,
save_path=save_path,
nimg_per_epoch=8,
learning_rate = self.training_params['learning_rate'],
Expand Down

0 comments on commit 9aff434

Please sign in to comment.