损失函数Target size must be the same as input size

出错程序 criterion_modality = torch.nn.BCEWithLogitsLoss() label = Variable(label.cuda()) loss = criterion_modality(outRGB, label) # 出错行 问题 File "C:\Users\Rain\AppData\Local\Programs\Python\Anaconda.3.5
相关文章
相关标签/搜索