我可以把train.py文件给成多分类码，我需要改哪些文件相应的参数，谢谢

JCruan519 / VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".

Apache License 2.0

517 stars 24 forks source link

Open sky1talent opened 5 months ago

PixDeep commented 4 months ago

可以参考 synapse 数据集，见 train_synapse.py 和 config/config_setting_synapse.py