From 8b22349f880b4d1d68acbbffc3fc8334e718d84a Mon Sep 17 00:00:00 2001 From: kapoorlab Date: Sun, 10 Nov 2024 20:55:36 +0000 Subject: [PATCH] version bump --- src/napatrackmater/Trackvector.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/napatrackmater/Trackvector.py b/src/napatrackmater/Trackvector.py index 426762f..5afc522 100644 --- a/src/napatrackmater/Trackvector.py +++ b/src/napatrackmater/Trackvector.py @@ -3021,7 +3021,8 @@ def train_gbr_vision_neural_net( experiment_name="mitosis", scheduler_choice="plateau", oneat_accuracy = True, - crop_size = None + crop_size = None, + pool_first = True ): @@ -3042,7 +3043,7 @@ def train_gbr_vision_neural_net( mitosis_inception.setup_gbr_vision_h5_datasets(crop_size = crop_size) - mitosis_inception.setup_densenet_vision_model(input_shape, num_classes,box_vector,start_kernel,mid_kernel,startfilter,depth, growth_rate) + mitosis_inception.setup_densenet_vision_model(input_shape, num_classes,box_vector,start_kernel,mid_kernel,startfilter,depth, growth_rate, pool_first = pool_first) mitosis_inception.setup_logger() mitosis_inception.setup_checkpoint()