ziyangwang007/Mamba-UNet

what is model1 ?

Qu1n-22 opened this issue · 3 comments

In the file 'train_Semi_Mamba_UNet.py'.
image

In the line 153-159 of the file 'train_Semi_Mamba_UNet.py'.
I change "model2" to "model1" in the line 153 and 155. I am not sure if this is correct, but the code works.

hi, please try model1 as unet, and model2 as mambaunet.

I'm sorry, I don't quite follow you, do I?