search

ziyangwang007 / Mamba-UNet

Mamba-UNet Zoo
Apache License 2.0
343 stars 22 forks source link

Something wrong in train_Semi_Mamba_UNet.py #25

Open onion-LHJ opened 1 week ago

onion-LHJ commented 1 week ago

When I ran train_Semi_Mamba_UNet.py, there is a bug: Traceback (most recent call last): File "train_Semi_Mamba_UNet.py", line 396, in train(args, snapshot_path) File "train_Semi_Mamba_UNet.py", line 183, in train model1.train() NameError: name 'model1' is not defined

And then I check the code ,notice that : 微信截图_20240625152742

I would be most grateful.if u could tell me what model1 should be .Thanks again for ur great work.

xskdjs444 commented 4 days ago

Hi, are you sloved this problem? this problem is caused by two model2 are definitioned. So you can ues function of create_model to definition model1 such as UNet . Hope this will help you!