你好，想问一下，在测试unet_proto模型训练出来的模型时，在test.py中的关于分类器的选择的代码是怎样的。

[weng-hb]

将你的源代码改为 elif args.ulp_classifier: outputs = net(data_bat) logits = outputs[0] compact_pred = torch.argmax(logits, dim=1) compact_pred = compact_pred.data.cpu().numpy() out = compact_pred[0,:,:].copy() 虽然能跑了，但是输出的结果不行。 大佬教教小白吧

[weng-hb]

包括使用你给的模型也是这样

[xjiangmed]

Sorry, I forgot to update the test code. I will check and re-upload the test code as soon as possible.

[weng-hb]

Thank you very much!哥，给你磕了

[weng-hb]

呜呜呜，谢谢啦哥！十分感谢！

[LeviAov]

呜呜呜，谢谢啦哥！十分感谢！

你好，请问你复现成功了吗？这个代码

[weng-hb]

呜呜呜，谢谢啦哥！十分感谢！

好，请问您复现成功了吗？

可以复现，但是我之前复现的时候训练和测试用的是两个环境，过了两个月了，具体细节记不太清了

[weng-hb]

呜呜呜，谢谢啦哥！十分感谢！

你好，请问你复现成功了吗？这个代码