tianrun-chen / xLSTM-UNet-PyTorch

Replacing Mamba with xLSTM! It works better. We show that xLSTM-Unet can be an effective semantic segmentation backbone.
123 stars 11 forks source link

Hello, I follow the steps why does this error occur #12

Open liubaoning111 opened 1 month ago

liubaoning111 commented 1 month ago

nnUNetv2_train DATASET_ID 3d_fullres all -tr nnUNetTrainerUxLSTMBot -lr {learning_rate} -bs {batch_size} usage: nnUNetv2_train [-h] [-tr TR] [-p P] [-pretrained_weights PRETRAINED_WEIGHTS] [-num_gpus NUM_GPUS] [--use_compressed] [--npz] [--c] [--val] [--val_best] [--disable_checkpointing] [-device DEVICE] dataset_name_or_id configuration fold nnUNetv2_train: error: unrecognized arguments: -lr {learning_rate} -bs {batch_size}