about batch_size about mednext

[saikat-roy]

There are different places to set the batch size. You could open the plans file in Python as in here and change it or you could add it to the constructor of any trainer in ./nnunet_mednext/training/network_training/MedNeXt/nnUNetTrainerV2_MedNeXt.py like so

def __init__(self, *args, **kwargs):
    super().__init__(*args, **kwargs)
    self.batch_size = 2

[CLGRY]

There are different places to set the batch size. You could open the plans file in Python as in here and change it or you could add it to the constructor of any trainer in ./nnunet_mednext/training/network_training/MedNeXt/nnUNetTrainerV2_MedNeXt.py like so

def __init__(self, *args, **kwargs):
    super().__init__(*args, **kwargs)
    self.batch_size = 2

May I ask you where the batch_size of this project can be changed, or what are the command line parameters, which can not be found at all.