train_Semi_Mamba_UNet

[W-L44]

NameError: name 'model1' is not defined. Did you mean: 'model2'?
请问，我按照一步一步来测试，出现这个问题应该如何解决，或者说对哪里进行改动，请告知

[Beatrice-kk]

model2 重复定义了 猜测是笔误

def train(args, snapshot_path):
base_lr = args.base_lr
num_classes = args.num_classes
batch_size = args.batch_size
max_iterations = args.max_iterations

def create_model(ema=False):

# Network definition

model = net_factory(net_type='unet', in_chns=1,

class_num=num_classes)

if ema:

for param in model.parameters():

param.detach_()

return model

#定义model1
model1 = ViM_seg(config, img_size=args.patch_size,
num_classes=args.num_classes).cuda()
model1.load_from(config)

#重复定义

model2 = ViM_seg(config, img_size=args.patch_size,

num_classes=args.num_classes).cuda()

model2.load_from(config)

model2 = ViM_seg(config, img_size=args.patch_size,
num_classes=args.num_classes).cuda()
model2.load_from(config)