可以用于多分类吗

[yi-coder18]

VM_Unet有用于二分类和多分类的图片的不同训练代码，请问UItraLight-VM-Unet有吗

[wurenkai]

Hi, the multi-targeting task is of course fine. You just need to replace 'vmunet.py' in the model folder with 'UItraLight-VM-UNet.py' with the help of VM-UNet's multi-targeting framework (train_synapse.py) '. Also, and modify the output 'return torch.sigmoid(out0)' to 'return out0' to run the multi-target task!