NickWang1122

你好,我也遇到了这个问题, model_output_one_hot = torch.nn.functional.one_hot(labels, num_classes=hp.out_class+1).permute(0,4,1,2,3)是这行代码报错 请问解决了吗？

按照您源代码的想法应该是得到所有样本数*155张切片送入网络，但是实际上送入网络的只有样本数个数的数据，应该是后面的len函数覆盖了前面len(self. Database) * 155